nn_jax

Neural Networks from scratch using JAX

Getting Started

git clone https://github.com/nn-jax/nn-jax.git
cd nn-jax
uv sync

Resources

• Deep Learning - Ian Goodfellow, Yoshua Bengio, Aaron Courville
• Neural Networks from Scratch - Harrison Kinsley, Daniel Kukiela
• Neural Network from Scratch - Omar Aflak
• MIT 6.S191 Introduction to Deep Learning - Alexander Amini and Ava Amini