lucasdelimanogueira / PyNorchView external linksLinks
Recreating PyTorch from scratch (C/C++, CUDA, NCCL and Python, with multi-GPU support and automatic differentiation!)
☆163Nov 25, 2025Updated 2 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.☆29Updated this week
- ☆11Apr 24, 2025Updated 9 months ago
- Row-wise block scaling for fp8 quantization matrix multiplication. Solution to GPU mode AMD challenge.☆17Feb 9, 2026Updated last week
- NVSHMEM‑Tutorial: Build a DeepEP‑like GPU Buffer☆163Updated this week
- Fast GPU based tensor core reductions☆13Jan 13, 2023Updated 3 years ago
- ☆46May 24, 2025Updated 8 months ago
- ☆19Jan 16, 2025Updated last year
- 🧠 A study guide to learn about Transformers☆12Jan 11, 2024Updated 2 years ago
- UNet diffusion model in pure CUDA☆661Jun 28, 2024Updated last year
- Lidar Obstacle Detector☆24Jun 3, 2025Updated 8 months ago
- High Performance FP8 GEMM Kernels for SM89 and later GPUs.☆20Jan 24, 2025Updated last year
- Soccer toy example simulator used in Reinforcement Learning☆12Mar 11, 2018Updated 7 years ago
- The first comprehensive Lean 4 formalization of statistical learning theory, featuring Gaussian Lipschitz concentration and Dudley's entr…☆30Feb 9, 2026Updated last week
- Machine translation with tinygrad☆19Apr 7, 2024Updated last year
- ☆12Jun 10, 2023Updated 2 years ago
- High-Performance FP32 GEMM on CUDA devices☆117Jan 21, 2025Updated last year
- High Performance Int8 GEMM Kernels for SM80 and later GPUs.☆19Mar 11, 2025Updated 11 months ago
- Open deep learning compiler stack for cpu, gpu and specialized accelerators☆19Feb 9, 2026Updated last week
- Trained a 114 million Parameter LLM from Scratch.☆19Jul 21, 2024Updated last year
- ☆13Jan 16, 2025Updated last year
- My submission for the GPUMODE/AMD fp8 mm challenge☆29Jun 4, 2025Updated 8 months ago
- Implementation of FlashAttention in PyTorch☆180Jan 12, 2025Updated last year
- PyTorch compilation tutorial covering TorchScript, torch.fx, and Slapo☆17Mar 13, 2023Updated 2 years ago
- A Easy-to-understand TensorOp Matmul Tutorial☆410Updated this week
- High performance NCCL plugin for Bagua.☆15Sep 15, 2021Updated 4 years ago
- flash attention tutorial written in python, triton, cuda, cutlass☆486Jan 20, 2026Updated 3 weeks ago
- A PyTorch implementation of the GPT-OSS-20B architecture. All components are coded from scratch: RoPE with YaRN, RMSNorm, SwiGLU with cla…☆209Dec 2, 2025Updated 2 months ago
- CUDA tutorials for Maths & ML tutorials with examples, covers multi-gpus, fused attention, winograd convolution, reinforcement learning.☆210Jun 11, 2025Updated 8 months ago
- Alex Krizhevsky's original code from Google Code☆199Mar 10, 2016Updated 9 years ago
- ☆93Jul 5, 2024Updated last year
- Paper implementation☆45Apr 8, 2025Updated 10 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.☆674Feb 4, 2026Updated last week
- ☆30Updated this week
- TACOS: [T]opology-[A]ware [Co]llective Algorithm [S]ynthesizer for Distributed Machine Learning☆31Jun 13, 2025Updated 8 months ago
- A bunch of kernels that might make stuff slower 😉☆75Updated this 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
- Minimalistic 4D-parallelism distributed training framework for education purpose☆2,076Aug 26, 2025Updated 5 months ago