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