Search results

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
projects 1 - 1 of 1
Share your project
Share what you've made with ML. Post Project
Learn practical AI
Learn ML with clean code, simplified math and visuals.
Lessons MWML lessons are among the top 10 ML repositories on GitHub.
Join the community
Get feedback on your projects, interview prep, and more! Chat on Slack