Projects

latest | popular

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
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
projects 1 - 3 of 3
Topic experts
Share a project
Share something interesting you found that's made with ML.