Projects

latest | popular

Filter by
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
Elegy
A Keras-like Deep Learning framework based on Jax + Haiku.
jax haiku keras article
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
Foolbox Native
A Python toolbox to create adversarial examples that fool neural networks in PyTorch, TensorFlow, and JAX
adversarial-learning adversarial-attacks pytorch tensorflow
JAX: Accelerated Machine Learning Research
This talk will introduce JAX and its core function transformations with a live demo.
jax video scipy-2020 tutorial
Solving Optimization Problems with JAX
JAX can be used to solve a range of simple to complex optimization problems with matrix methods.
jax autograd xla optimization
SymJAX
A symbolic CPU/GPU/TPU programming
jax xla autograd symjax
Convoluted Stuff
Optimising compilers, and how thousand-year-old math shaped deep learning.
matrix-multiplication convolutional-neural-networks jax im2col
Implementing Graph Neural Networks with JAX
I’ll talk about my experience on how to build and train Graph Neural Networks (GNNs) with JAX.
graph-neural-networks jax graphs tutorial
projects 1 - 10 of 16
Topic experts
Share a project
Share something you or the community has made with ML.