Check out the
topics page for highly curated tutorials and libraries on jax.

2020-03-29 · 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

2020-03-26 · A gentle introduction to Flax: a neural network library for JAX that is designed for flexibility.

flax jax deep-learning library

A Python toolbox to create adversarial examples that fool neural networks in PyTorch, TensorFlow, and JAX

adversarial-learning adversarial-attacks pytorch tensorflow

2020-07-05 · This talk will introduce JAX and its core function transformations with a live demo.

jax video scipy-2020 tutorial

2020-05-25 · JAX can be used to solve a range of simple to complex optimization problems with matrix methods.

jax autograd xla optimization

2020-05-17 · Optimising compilers, and how thousand-year-old math shaped deep learning.

matrix-multiplication convolutional-neural-networks jax im2col

2020-04-20 · 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**