Projects

latest | popular

Filter by
Finetuning Transformers with JAX + Haiku
Walking through a port of the RoBERTa pre-trained model to JAX + Haiku, then fine-tuning the model to solve a downstream task.
jax haiku roberta transformers
From PyTorch to JAX
Towards neural net frameworks that purify stateful code.
jax haiku tutorial article
projects 1 - 2 of 2
Topic experts
Share your project
Discover, build and share what you've made with ML.
Share someone else's project
Share something interesting you found that's made with ML.