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

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

2020-03-09 · Towards neural net frameworks that purify stateful code.

jax haiku tutorial

Trying to learn a simulation? Try Lagrangian Neural Networks, which explicitly conserve energy and may generalize better!

deep-learning graph-neural-networks jax interpretability

2020-03-15 · Optimizing the filters to improve the filtered images for computer vision tasks.

jax numpy computer-vision separable-filters

projects **1 - 9** of **9**