StijnWoestenborghs / gradi-mojoLinks
☆38Updated 2 years ago
Alternatives and similar repositories for gradi-mojo
Users that are interested in gradi-mojo are comparing it to the libraries listed below
Sorting:
- A Learning Journey: Micrograd in Mojo 🔥☆63Updated last year
- Implementation of Karpathy's micrograd in Mojo☆78Updated 2 years ago
- ☆53Updated 4 months ago
- Machine Learning library for the emerging Mojo/Python ecosystem☆297Updated 3 weeks ago
- Contains supporting materials for developer relations blog posts, videos, and workshops☆46Updated last month
- A package for defining deep learning models using categorical algebraic expressions.☆61Updated last year
- A fast and compact Dict implementation in Mojo 🔥☆36Updated 4 months ago
- A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.☆297Updated last year
- ☆28Updated last year
- Notebooks for the "Deep Learning with JAX" book☆162Updated 6 months ago
- Rust Implementation of micrograd☆53Updated last year
- Your favourite classical machine learning algos on the GPU/TPU☆20Updated 11 months ago
- A working machine learning framework in pure Mojo 🔥☆130Updated last year
- ☆112Updated 2 years ago
- NanoGPT (124M) in 5 minutes☆14Updated 10 months ago
- Jax like function transformation engine but micro, microjax☆33Updated last year
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆125Updated 2 months ago
- Neural Networks for JAX☆84Updated last year
- Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`☆47Updated last year
- ☆28Updated last year
- This code repository contains the code used for my "Optimizing Memory Usage for Training LLMs and Vision Transformers in PyTorch" blog po…☆92Updated 2 years ago
- Explorations into the proposal from the paper "Grokfast, Accelerated Grokking by Amplifying Slow Gradients"☆103Updated 11 months ago
- Minimal GPT (~350 lines with a simple task to test it)☆63Updated 2 weeks ago
- Swarming algorithms like PSO, Ant Colony, Sakana, and more in PyTorch 😊☆134Updated last month
- NanoGPT (124M) quality in 2.67B tokens☆28Updated 2 months ago
- A Mojo implementation of the Tiny Stable Diffusion model☆56Updated last year
- Documented and Unit Tested educational Deep Learning framework with Autograd from scratch.☆122Updated last year
- Graph neural networks in JAX.☆68Updated last year
- ☆153Updated last month
- Kolmogorov-Arnold Networks in Mojo☆19Updated 4 months ago