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
A symbolic CPU/GPU/TPU programming
jax xla autograd symjax
A tiny scalar-valued autograd engine and a neural net library on top of it with PyTorch-like API
autograd neural-networks code library
projects 1 - 5 of 5
Topic experts
Share a project
Share something you or the community has made with ML.