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
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
projects 1 - 4 of 4
Topic experts
Share a project
Share something you or the community has made with ML.