dnbt777 / jax-from-scratchLinks
ML from scratch in Jax
☆12Updated 2 months ago
Alternatives and similar repositories for jax-from-scratch
Users that are interested in jax-from-scratch are comparing it to the libraries listed below
Sorting:
- small auto-grad engine inspired from Karpathy's micrograd and PyTorch☆275Updated 10 months ago
- could we make an ml stack in 100,000 lines of code?☆46Updated last year
- in this repository, i'm going to implement increasingly complex llm inference optimizations☆68Updated 4 months ago
- Learnings and programs related to CUDA☆420Updated 3 months ago
- Following master Karpathy with GPT-2 implementation and training, writing lots of comments cause I have memory of a goldfish☆172Updated last year
- ☆509Updated 2 months ago
- Solve puzzles to improve your tinygrad skills!☆145Updated this week
- ☆92Updated 8 months ago
- High Quality Resources on GPU Programming/Architecture☆589Updated last year
- (WIP) A small but powerful, homemade PyTorch from scratch.☆643Updated this week
- Tutorials on tinygrad☆419Updated last week
- This repo has all the basic things you'll need in-order to understand complete vision transformer architecture and its various implementa…☆228Updated 9 months ago
- a tiny multidimensional array implementation in C similar to numpy, but only one file.☆228Updated last year
- Atropos is a Language Model Reinforcement Learning Environments framework for collecting and evaluating LLM trajectories through diverse …☆714Updated this week
- a tiny vectorstore implementation built with numpy.☆63Updated last year
- ☆56Updated 6 months ago
- rl from zero pretrain, can it be done? yes.☆275Updated 3 weeks ago
- Copy code from your codebase to clipboard instantly for LLM context!☆324Updated 2 months ago
- go the distance☆284Updated 4 months ago
- A curated list of resources for learning and exploring Triton, OpenAI's programming language for writing efficient GPU code.☆421Updated 7 months ago
- My path from leetcode hell to bigtech heaven☆16Updated last year
- Simple Transformer in Jax☆139Updated last year
- A deep dive on the history of robotics and the future of humanoids☆108Updated 10 months ago
- ☆94Updated last week
- Async RL Training at Scale☆709Updated this week
- adammaj.com | Built with Next & Chakra☆91Updated last week
- control your music from the command line☆143Updated 5 months ago
- Quantized LLM training in pure CUDA/C++.☆198Updated last week
- ☆96Updated last year
- SIMD quantization kernels☆87Updated last month