Recreating PyTorch from scratch (C/C++, CUDA, NCCL and Python, with multi-GPU support and automatic differentiation!)
☆168Nov 25, 2025Updated 8 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:
- ☆46May 24, 2025Updated last year
- Machine translation with tinygrad☆19Apr 7, 2024Updated 2 years ago
- UNet diffusion model in pure CUDA☆663Jun 28, 2024Updated 2 years ago
- This repo implements and trains DallE-1 on a synthetically generated dataset which has colored mnist images on texture/solid background a…☆14Oct 30, 2024Updated last year
- FalkorDB port to Rust☆13Mar 26, 2026Updated 4 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.
- Contrastive Reinforcement Learning☆67Apr 4, 2026Updated 4 months ago
- Implementation of FlashAttention (FA1-FA4) in PyTorch for educational and algorithmic clarity☆223Apr 12, 2026Updated 4 months ago
- Conformer block with Rotary Position Embedding, modified from lucidrains' implement☆19Sep 13, 2024Updated last year
- 53 implementations of synthetic learning problems from Geoffrey Hinton's experimental papers (1981-2022). Pure numpy, laptop-runnable, pa…☆33May 11, 2026Updated 3 months ago
- 6th place solution☆22Feb 28, 2023Updated 3 years ago
- Minimalistic 4D-parallelism distributed training framework for education purpose☆2,282Aug 26, 2025Updated 11 months ago
- Optimize tensor program fast with Felix, a gradient descent autotuner.☆33Mar 5, 2026Updated 5 months ago
- Comprehensive CUDA tutorials for Maths & ML with examples☆237Jun 11, 2025Updated last year
- Row-wise block scaling for fp8 quantization matrix multiplication. Solution to GPU mode AMD challenge.☆19Feb 9, 2026Updated 6 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.
- Toolkit for morse code audio generation and training DNNs for automatic morse code recognition☆11Aug 5, 2022Updated 4 years ago
- Open deep learning compiler stack for cpu, gpu and specialized accelerators☆20Aug 5, 2026Updated 2 weeks ago
- High Performance FP8 GEMM Kernels for SM89 and later GPUs.☆21Jan 24, 2025Updated last year
- High-Performance FP32 GEMM on CUDA devices☆126Jan 21, 2025Updated last year
- 使用 CUDA C++ 实现的 llama 模型推理框架☆64Nov 8, 2024Updated last year
- This is a small autograd engine, made purely from numpy and python.☆27Sep 17, 2024Updated last year
- JCudnn - Java bindings for cuDNN☆31Nov 16, 2024Updated last year
- C Compiler written in Kotlin☆13Apr 19, 2024Updated 2 years ago
- A Easy-to-understand TensorOp Matmul Tutorial☆451Mar 5, 2026Updated 5 months ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- IMMA-based **FP8-as-storage** GEMM experiments for Ampere (sm_86 / RTX 3090 Ti).☆24Jan 30, 2026Updated 6 months ago
- 使用 cutlass 实现 flash-attention 精简版，具有教学意义☆59Aug 12, 2024Updated 2 years ago
- flash attention tutorial written in python, triton, cuda, cutlass☆532Jan 20, 2026Updated 6 months ago
- rl from zero pretrain, can it be done? yes.☆295Sep 28, 2025Updated 10 months ago
- Find, list, and inspect processes from Go (golang).☆10Feb 4, 2018Updated 8 years ago
- A PyTorch implementation of the GPT-OSS-20B architecture. All components are coded from scratch: RoPE with YaRN, RMSNorm, SwiGLU with cla…☆238Dec 2, 2025Updated 8 months ago
- ☆29Apr 25, 2025Updated last year
- A C11 compiler for the discrete logic computer☆21Apr 3, 2024Updated 2 years ago
- A zero-dependency ML framework in C with a modern Python API for full control over execution and memory.☆700Updated this week
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Implement custom operators in PyTorch with cuda/c++☆77Jan 1, 2023Updated 3 years ago
- ☆17Jan 22, 2023Updated 3 years ago
- Code for ACL22 short Paper "Hierarchical Curriculum Learning for AMR Parsing"☆13Jun 1, 2022Updated 4 years ago
- Experimenting with small language models☆75Jan 16, 2024Updated 2 years ago
- Yet Another Language Model: LLM inference in C++/CUDA, no libraries except for I/O☆595Sep 13, 2025Updated 11 months ago
- Repository for the COLM 2025 paper SpecDec++: Boosting Speculative Decoding via Adaptive Candidate Lengths☆19Jul 10, 2025Updated last year
- AI powered tool for Git commit message generator☆11Jul 29, 2025Updated last year