MarioSieg / magnetron
(WIP) A small but powerful, homemade PyTorch from scratch.
☆482Updated this week
Alternatives and similar repositories for magnetron:
Users that are interested in magnetron are comparing it to the libraries listed below
- small auto-grad engine inspired from Karpathy's micrograd and PyTorch☆232Updated last month
- This repo has all the basic things you'll need in-order to understand complete vision transformer architecture and its various implementa…☆192Updated 2 weeks ago
- The Tensor (or Array)☆418Updated 5 months ago
- The Autograd Engine☆550Updated 4 months ago
- UNet diffusion model in pure CUDA☆596Updated 6 months ago
- A curated list of resources for learning and exploring Triton, OpenAI's programming language for writing efficient GPU code.☆161Updated this week
- High Quality Resources on GPU Programming/Architecture☆577Updated 5 months ago
- Best practices & guides on how to write distributed pytorch training code☆336Updated this week
- The Multilayer Perceptron Language Model☆532Updated 5 months ago
- Solve puzzles to improve your tinygrad skills!☆100Updated 3 months ago
- Following master Karpathy with GPT-2 implementation and training, writing lots of comments cause I have memory of a goldfish☆167Updated 5 months ago
- An ML Systems Onboarding list☆647Updated 2 months ago
- pytorch from scratch in pure C/CUDA and python☆39Updated 3 months ago
- a tiny multidimensional array implementation in C similar to numpy, but only one file.☆225Updated 5 months ago
- Learnings and programs related to CUDA☆95Updated this week
- This is the official repository for The Hundred-Page Language Models Book by Andriy Burkov☆195Updated last week
- gpt-2 from scratch in mlx☆367Updated 7 months ago
- Minimalistic 4D-parallelism distributed training framework for education purpose☆644Updated this week
- High-Performance FP32 Matrix Multiplication on CPU☆325Updated 3 weeks ago
- NanoGPT (124M) in 3.4 minutes☆2,068Updated last week
- Simple Byte pair Encoding mechanism used for tokenization process . written purely in C☆122Updated 2 months ago
- Solve Puzzles. Learn Metal 🤘☆497Updated 3 months ago
- Puzzles for learning Triton☆1,300Updated last month
- noise_step: Training in 1.58b With No Gradient Memory☆214Updated 3 weeks ago
- Alex Krizhevsky's original code from Google Code☆190Updated 8 years ago
- learningggggggg 🐳☆130Updated last week
- port of Andrjey Karpathy's llm.c to Mojo☆349Updated last month
- ☆413Updated 2 months ago
- A reinforcement learning framework based on MLX.☆222Updated 11 months ago
- Implementation of Diffusion Transformer (DiT) in JAX☆261Updated 7 months ago