This module defines a type system for distributed training code, based off of JAX's sharding in types, but adapted for the PyTorch ecosystem.
☆34Jul 7, 2026Updated last week
Alternatives and similar repositories for spmd_types
Users that are interested in spmd_types are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- My attempt to improve the speed of the newton schulz algorithm, starting from the dion implementation.☆38Apr 30, 2026Updated 2 months ago
- Implementation of FixMatch in PyTorch and experimentations☆12Aug 9, 2020Updated 5 years ago
- ☆17Apr 1, 2020Updated 6 years ago
- Fork of Triton repository for OpenXLA uses of the Triton language and compiler☆16Feb 24, 2026Updated 4 months ago
- Vine: The BitBlaze Static Analysis Component☆26Sep 27, 2014Updated 11 years ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- GPU Affinity is a package to automatically set the CPU process affinity to match the hardware architecture on a given platform☆29Dec 8, 2023Updated 2 years ago
- gradient-based symbolic execution engine implemented from scratch☆38Dec 16, 2023Updated 2 years ago
- An experimental implementation of compiler-driven automatic sharding of models across a given device mesh.☆89Updated this week
- ☆65Dec 10, 2025Updated 7 months ago
- Make triton easier☆49Jun 12, 2024Updated 2 years ago
- Automatically insert nvtx ranges to PyTorch models☆22Apr 28, 2021Updated 5 years ago
- Compact and Agent-Native MoE Training System☆253Updated this week
- Mechanized baselines for various type system features☆19Apr 14, 2026Updated 3 months ago
- A synthetic story narration dataset to study small audio LMs.☆31Jan 21, 2024Updated 2 years 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.
- My study note for mlsys☆14Nov 4, 2024Updated last year
- Reference implementation for "Group Equivariant Subsampling"☆16Dec 13, 2021Updated 4 years ago
- Manage ML configuration with pydantic☆16Mar 18, 2026Updated 3 months ago
- Companion website of "MorphQ: Metamorphic Testing of the Qiskit Quantum Computing Platform", accepted at ICSE 23☆10Mar 12, 2025Updated last year
- ☆170Dec 27, 2024Updated last year
- Standalone commandline CLI tool for compiling Triton kernels☆20Sep 13, 2024Updated last year
- ☆16Apr 11, 2024Updated 2 years ago
- An Apache 2.0 fork of HuggingFace's Large Language Model Text Generation Inference☆19Mar 10, 2024Updated 2 years ago
- Glacier is a type annotation extension to Java that enforces transitive class immutability.☆15Nov 15, 2021Updated 4 years ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- library which simplifies host-GPU data transfer using userspace pagefault handling☆15Jun 8, 2012Updated 14 years ago
- Infiniband verbs performance tests (fork of git://git.openfabrics.org/~grockah/perftest.git)☆21Jan 14, 2016Updated 10 years ago
- Bin2Wrong: a Unified Fuzzing Framework for Uncovering Semantic Errors in Binary-to-C Decompilers☆61May 20, 2025Updated last year
- ☆79May 4, 2021Updated 5 years ago
- Code for GFlowNet-DPO (Direct Preference Optimization) EMNLP 2024 Main☆19Feb 22, 2026Updated 4 months ago
- [ICML 2025 Spotlight] ShadowKV: KV Cache in Shadows for High-Throughput Long-Context LLM Inference☆308May 1, 2025Updated last year
- ☆14Jun 26, 2024Updated 2 years ago
- SIMD quantization kernels☆93May 29, 2026Updated last month
- ☆18Oct 15, 2020Updated 5 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- ☆24Apr 8, 2026Updated 3 months ago
- ☆38Jun 27, 2025Updated last year
- ☆192Jun 16, 2024Updated 2 years ago
- Drop-in library for tracking the memory allocations of CUDA applications☆14Nov 17, 2017Updated 8 years ago
- Observe the dataset of images and targets in few shots☆11Sep 27, 2022Updated 3 years ago
- [NeurIPS 2025] Official implementation of the paper "BecomingLit: Relightable Gaussian Avatars with Hybrid Neural Shading"☆33Jun 10, 2026Updated last month
- mKernel: fast multi-node, multi-GPU fused kernels☆251Jun 21, 2026Updated 3 weeks ago