Recreating PyTorch from scratch (C/C++, CUDA, NCCL and Python, with multi-GPU support and automatic differentiation!)
☆164Nov 25, 2025Updated 3 months ago
Alternatives and similar repositories for PyNorch
Users that are interested in PyNorch are comparing it to the libraries listed below
Sorting:
- A std::execution style runtime context and High Performance RPC Transport for using OpenUCX. Including CUDA/ROCM/... devices with RDMA.☆29Feb 22, 2026Updated 2 weeks ago
- Row-wise block scaling for fp8 quantization matrix multiplication. Solution to GPU mode AMD challenge.☆18Feb 9, 2026Updated last month
- NVSHMEM‑Tutorial: Build a DeepEP‑like GPU Buffer☆167Feb 11, 2026Updated last month
- A simple implementation of Llama 1, 2. Llama Architecture built from scratch using PyTorch all the models are built from scratch that inc…☆14May 6, 2024Updated last year
- SpeechPlus: Small LLM-Based Text-to-Speech Library 🚀☆20May 20, 2025Updated 9 months ago
- Fast GPU based tensor core reductions☆13Jan 13, 2023Updated 3 years ago
- ☆46May 24, 2025Updated 9 months ago
- Conformer block with Rotary Position Embedding, modified from lucidrains' implement☆18Sep 13, 2024Updated last year
- 🧠 A study guide to learn about Transformers☆12Jan 11, 2024Updated 2 years ago
- OpenMM Metal Plugin☆19Aug 20, 2024Updated last year
- UNet diffusion model in pure CUDA☆657Jun 28, 2024Updated last year
- Soccer toy example simulator used in Reinforcement Learning☆12Mar 11, 2018Updated 8 years ago
- High Performance FP8 GEMM Kernels for SM89 and later GPUs.☆20Jan 24, 2025Updated last year
- Lidar Obstacle Detector☆24Jun 3, 2025Updated 9 months ago
- Open deep learning compiler stack for cpu, gpu and specialized accelerators☆19Feb 24, 2026Updated 2 weeks ago
- High Performance Int8 GEMM Kernels for SM80 and later GPUs.☆20Mar 11, 2025Updated last year
- My submission for the GPUMODE/AMD fp8 mm challenge☆29Jun 4, 2025Updated 9 months ago
- ☆13Jan 16, 2025Updated last year
- 📑 Dive into Big Model Training☆115Dec 1, 2022Updated 3 years ago
- Implementation of FlashAttention in PyTorch☆182Jan 12, 2025Updated last year
- Implement custom operators in PyTorch with cuda/c++☆76Jan 1, 2023Updated 3 years ago
- The first comprehensive Lean 4 formalization of statistical learning theory, featuring Gaussian Lipschitz concentration and Dudley's entr…☆55Mar 5, 2026Updated last week
- A Easy-to-understand TensorOp Matmul Tutorial☆408Mar 5, 2026Updated last week
- High performance NCCL plugin for Bagua.☆15Sep 15, 2021Updated 4 years ago
- flash attention tutorial written in python, triton, cuda, cutlass☆490Jan 20, 2026Updated last month
- Comprehensive CUDA tutorials for Maths & ML with examples.☆215Jun 11, 2025Updated 9 months ago
- Alex Krizhevsky's original code from Google Code☆199Mar 10, 2016Updated 10 years ago
- CFR implementation of a poker bot.☆12Feb 17, 2023Updated 3 years ago
- ☆93Jul 5, 2024Updated last year
- RL-Scope: Cross-Stack Profiling for Deep Reinforcement Learning Workloads☆47Apr 7, 2021Updated 4 years ago
- A PyTorch implementation of the GPT-OSS-20B architecture. All components are coded from scratch: RoPE with YaRN, RMSNorm, SwiGLU with cla…☆219Dec 2, 2025Updated 3 months ago
- Paper implementation☆48Apr 8, 2025Updated 11 months ago
- A simple Python tool to measure the performance of ONNX models.☆27Sep 15, 2024Updated last year
- (WIP) A small but powerful, homemade PyTorch from scratch.☆676Updated this week
- TACOS: [T]opology-[A]ware [Co]llective Algorithm [S]ynthesizer for Distributed Machine Learning☆32Jun 13, 2025Updated 9 months ago
- ☆31Updated this week
- A bunch of kernels that might make stuff slower 😉☆75Mar 2, 2026Updated last week
- Gensis is a lightweight deep learning framework written from scratch in Python, with Triton as its backend for high-performance computing…☆37Jan 15, 2026Updated last month
- Reference implementation of Mistral AI 7B v0.1 model.☆28Dec 25, 2023Updated 2 years ago