#

haiku-trainer

Here are 2 public repositories matching this topic...

NTT123 / pointer-networks

An unofficial implementation of pointer networks.

deep-learning pointer-networks pca convex-hull jax ptr-net dm-haiku haiku-trainer

Updated Dec 6, 2020
Jupyter Notebook

NTT123 / haiku_trainer

A helper library for training dm-haiku models.

machine-learning deep-learning trainer jax dm-haiku optax haiku-trainer

Updated Feb 7, 2021
Python

Improve this page

Add a description, image, and links to the haiku-trainer topic page so that developers can more easily learn about it.

Curate this topic

Add this topic to your repo

To associate your repository with the haiku-trainer topic, visit your repo's landing page and select "manage topics."