Search results

Getting started with JAX (MLPs, CNNs & RNNs)
Learn the building blocks of JAX and use them to build some standard Deep Learning architectures (MLP, CNN, RNN, etc.).
jax xla autograd tpu
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
Flax: Google’s Open Source Approach To Flexibility In ML
A gentle introduction to Flax: a neural network library for JAX that is designed for flexibility.
flax jax deep-learning library
From PyTorch to JAX
Towards neural net frameworks that purify stateful code.
jax haiku tutorial
Using JAX to Improve Separable Image Filters
Optimizing the filters to improve the filtered images for computer vision tasks.
jax numpy computer-vision separable-filters
projects 1 - 5 of 5
Share your project
Share what you've made with ML. Post Project
Learn practical AI
Learn ML with clean code, simplified math and visuals.
Lessons MWML lessons are among the top 10 ML repositories on GitHub.
Join the community
Get feedback on your projects, interview prep, and more! Chat on Slack