PABannier / nanogradLinks
A lightweight deep learning framework
☆34Updated 4 years ago
Alternatives and similar repositories for nanograd
Users that are interested in nanograd are comparing it to the libraries listed below
Sorting:
- Symbolic API for model creation in PyTorch.☆66Updated 4 months ago
- The boundary of neural network trainability is fractal☆215Updated last year
- Documented and Unit Tested educational Deep Learning framework with Autograd from scratch.☆120Updated last year
- Automatic gradient descent☆208Updated 2 years ago
- ☆275Updated last year
- A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.☆290Updated 11 months ago
- A tool to analyze and debug neural networks in pytorch. Use a GUI to traverse the computation graph and view the data from many different…☆288Updated 8 months ago
- A miniscule implementation of reverse mode auto-differentiation☆29Updated 3 years ago
- ☆150Updated 11 months ago
- Unofficial JAX implementations of deep learning research papers☆156Updated 3 years ago
- Resources from the EleutherAI Math Reading Group☆53Updated 5 months ago
- Neural Networks for JAX☆84Updated 10 months ago
- A pure NumPy implementation of Mamba.☆223Updated last year
- Universal Tensor Operations in Einstein-Inspired Notation for Python.☆394Updated 4 months ago
- Hierarchical Associative Memory User Experience☆103Updated 3 weeks ago
- An interactive exploration of Transformer programming.☆267Updated last year
- A pure-functional implementation of a machine learning transformer model in Python/JAX☆179Updated 3 months ago
- A really tiny autograd engine☆95Updated 2 months ago
- Various handy scripts to quickly setup new Linux and Windows sandboxes, containers and WSL.☆40Updated 3 months ago
- A minimal implementation of autograd (in pure Python) 🍰☆95Updated 3 years ago
- Deep learning library implemented from scratch in numpy. Mixtral, Mamba, LLaMA, GPT, ResNet, and other experiments.☆50Updated last year
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆179Updated 2 weeks ago
- A repository for log-time feedforward networks☆223Updated last year
- This repository contain the simple llama3 implementation in pure jax.☆68Updated 5 months ago
- Functional local implementations of main model parallelism approaches☆96Updated 2 years ago
- Run PyTorch in JAX. 🤝☆268Updated this week
- This is a port of Mistral-7B model in JAX☆32Updated last year
- Named tensors with first-class dimensions for PyTorch☆332Updated 2 years ago
- Small deep learning library written from scratch in Python, using NumPy/CuPy.☆125Updated 2 years ago
- minGPT in JAX☆48Updated 3 years ago