Recreating PyTorch from scratch (C/C++, CUDA, NCCL and Python, with multi-GPU support and automatic differentiation!)
☆166Nov 25, 2025Updated 5 months ago
Alternatives and similar repositories for PyNorch
Users that are interested in PyNorch are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Contrastive Reinforcement Learning☆63Apr 4, 2026Updated last month
- SpeechPlus: Small LLM-Based Text-to-Speech Library 🚀☆21May 20, 2025Updated 11 months ago
- 🧠 A study guide to learn about Transformers☆12Jan 11, 2024Updated 2 years ago
- Machine translation with tinygrad☆19Apr 7, 2024Updated 2 years ago
- Alex Krizhevsky's original code from Google Code☆199Mar 10, 2016Updated 10 years ago
- Simple, predictable pricing with DigitalOcean hosting • AdAlways know what you'll pay with monthly caps and flat pricing. Enterprise-grade infrastructure trusted by 600k+ customers.
- ☆46May 24, 2025Updated 11 months ago
- UNet diffusion model in pure CUDA☆657Jun 28, 2024Updated last year
- This repo implements and trains DallE-1 on a synthetically generated dataset which has colored mnist images on texture/solid background a…☆13Oct 30, 2024Updated last year
- A std::execution style runtime context and High Performance RPC Transport for using OpenUCX. Including CUDA/ROCM/... devices with RDMA.☆30Updated this week
- 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 2 years ago
- High Performance FP8 GEMM Kernels for SM89 and later GPUs.☆21Jan 24, 2025Updated last year
- Lidar Obstacle Detector☆25Jun 3, 2025Updated 11 months ago
- High-Performance FP32 GEMM on CUDA devices☆125Jan 21, 2025Updated last year
- Implementation of FlashAttention (FA1-FA4) in PyTorch for educational and algorithmic clarity☆213Apr 12, 2026Updated last month
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Trained a 114 million Parameter LLM from Scratch.☆19Jul 21, 2024Updated last year
- Paper implementation☆53Apr 8, 2025Updated last year
- Conformer block with Rotary Position Embedding, modified from lucidrains' implement☆19Sep 13, 2024Updated last year
- Reference implementation of Mistral AI 7B v0.1 model.☆28Dec 25, 2023Updated 2 years ago
- ☆93Jul 5, 2024Updated last year
- Ecosystem for writing partial differential equation solvers☆15Dec 15, 2025Updated 5 months ago
- Machine learning algorithms implements with jax for machine learning in production in large scale dataset.☆16May 11, 2026Updated last week
- High Performance Int8 GEMM Kernels for SM80 and later GPUs.☆23Mar 11, 2025Updated last year
- Comprehensive CUDA tutorials for Maths & ML with examples☆224Jun 11, 2025Updated 11 months ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Optimize tensor program fast with Felix, a gradient descent autotuner.☆33Mar 5, 2026Updated 2 months ago
- Toolkit for morse code audio generation and training DNNs for automatic morse code recognition☆11Aug 5, 2022Updated 3 years ago
- A zero-to-one guide on scaling modern transformers with n-dimensional parallelism.☆124Dec 29, 2025Updated 4 months ago
- ☆10Jan 28, 2024Updated 2 years ago
- Open deep learning compiler stack for cpu, gpu and specialized accelerators☆19May 13, 2026Updated last week
- ☆12Apr 24, 2025Updated last year
- Fast GPU based tensor core reductions☆13Jan 13, 2023Updated 3 years ago
- ☆17Oct 5, 2024Updated last year
- A curated collection of prompts for Grok Imagine by xAI☆29Oct 19, 2025Updated 7 months ago
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- NVSHMEM‑Tutorial: Build a DeepEP‑like GPU Buffer☆190Feb 11, 2026Updated 3 months ago
- TACOS: [T]opology-[A]ware [Co]llective Algorithm [S]ynthesizer for Distributed Machine Learning☆34Jun 13, 2025Updated 11 months ago
- C Compiler written in Kotlin☆13Apr 19, 2024Updated 2 years ago
- JCudnn - Java bindings for cuDNN☆31Nov 16, 2024Updated last year
- 📑 Dive into Big Model Training☆115Dec 1, 2022Updated 3 years ago
- Implementation of a GPT-4o like Multimodal from Scratch using Python☆77Apr 4, 2025Updated last year
- Minimalistic 4D-parallelism distributed training framework for education purpose☆2,182Aug 26, 2025Updated 8 months ago