haiku-trainer

There are 2 repositories under haiku-trainer topic.

NTT123/pointer-networks
An unofficial implementation of pointer networks.
Language:Jupyter Notebook2 3 0
NTT123/haiku_trainer
A helper library for training dm-haiku models.
Language:Python1 3 0