modular / mojo-gpu-puzzlesLinks
Learn GPU Programming in Mojo🔥 by Solving Puzzles
☆82Updated last week
Alternatives and similar repositories for mojo-gpu-puzzles
Users that are interested in mojo-gpu-puzzles are comparing it to the libraries listed below
Sorting:
- Composable Function Transformations in Python with Mojo/MAX acceleration☆257Updated last week
- A Learning Journey: Micrograd in Mojo 🔥☆61Updated 9 months ago
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆115Updated 3 weeks ago
- Tensor library with autograd using only Rust's standard library☆68Updated last year
- Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs☆424Updated last week
- ☆28Updated 10 months ago
- port of Andrjey Karpathy's llm.c to Mojo☆353Updated last week
- Implementation of Karpathy's micrograd in Mojo☆74Updated last year
- PyTorch Single Controller☆318Updated this week
- Notebooks for the "Deep Learning with JAX" book☆150Updated last month
- A Jax-based library for building transformers, includes implementations of GPT, Gemma, LlaMa, Mixtral, Whisper, SWin, ViT and more.☆290Updated 10 months ago
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆280Updated this week
- Complete solutions to the Programming Massively Parallel Processors Edition 4☆196Updated 3 weeks ago
- A working machine learning framework in pure Mojo 🔥☆130Updated last year
- A Machine Learning framework from scratch in Pure Mojo 🔥☆443Updated 5 months ago
- Learning about CUDA by writing PTX code.☆133Updated last year
- C API for MLX☆117Updated this week
- Where GPUs get cooked 👩🍳🔥☆236Updated 4 months ago
- The Tensor (or Array)☆437Updated 11 months ago
- Orbax provides common checkpointing and persistence utilities for JAX users☆404Updated this week
- Modular, scalable library to train ML models☆135Updated this week
- Small scale distributed training of sequential deep learning models, built on Numpy and MPI.☆134Updated last year
- ☆54Updated last year
- ☆37Updated last year
- NuMojo is a library for numerical computing in Mojo 🔥 similar to numpy in Python.☆165Updated 2 weeks ago
- SIMD quantization kernels☆73Updated last week
- Simple MPI implementation for prototyping or learning☆259Updated 2 weeks ago
- A repository to unravel the language of GPUs, making their kernel conversations easy to understand☆188Updated last month
- Competitive GPU kernel optimization platform.☆82Updated last week
- ☆132Updated 2 weeks ago