Local and global SPMD type system supporting erasure
☆10Apr 24, 2026Updated 6 months ago
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:
- This module defines a type system for distributed training code, based off of JAX's sharding in types, but adapted for the PyTorch ecosys…☆46Updated this week
- Playwright-like automation framework for terminal TUI applications☆23Feb 10, 2026Updated 8 months ago
- torch_remat fine-grained activation checkpointing API☆23Updated this week
- jni bindings to aten from libtorch☆11Apr 20, 2025Updated last year
- An experimental implementation of compiler-driven automatic sharding of models across a given device mesh.☆97Oct 21, 2026Updated last week
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- Learn one, get them all for free☆12Jan 28, 2024Updated 2 years ago
- ☆10Sep 4, 2021Updated 5 years ago
- GPU-accelerated AES encryption project☆11Feb 13, 2015Updated 11 years ago
- Benchmarks to capture important workloads.☆33Apr 1, 2026Updated 6 months ago
- ☆11Jun 21, 2024Updated 2 years ago
- Skill pack that turns any coding agent (Claude Code, Codex, AGENTS.md-compatible) into a senior production engineer — FastAPI, PostgreSQL…☆26Apr 16, 2026Updated 6 months ago
- An AI that learns to play Super Smash Bros. Melee via imitation learning☆23Feb 22, 2024Updated 2 years ago
- SAEs in Jax☆12Dec 16, 2024Updated last year
- Skills for agents using Julia - general info and package / ecosystem specific skills!☆30Mar 7, 2026Updated 7 months ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- 100% MFU or bust.☆11Jan 12, 2026Updated 9 months ago
- Minimal example of using Tenstorrent TTNN in C++☆14Apr 12, 2025Updated last year
- Auto-differentiation library for C++☆12Jan 16, 2022Updated 4 years ago
- MXFP4-compatible 4-bit floating point types and block formats for Rust.☆12Mar 4, 2026Updated 7 months ago
- Easy no-frills Jax implementations of common abstractions for simple diffusion models.☆12Feb 23, 2026Updated 8 months ago
- Comparative study and experimentation on standard vs mHC vs attention residual (full and block)☆13Mar 17, 2026Updated 7 months ago
- Brain floating point type for Swift☆12Oct 12, 2025Updated last year
- Memory footprint reduction for transformer models☆11Jan 24, 2023Updated 3 years ago
- ☆14Nov 27, 2025Updated 11 months ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Analyzes whole genome sequencing data for gene-editing verification☆13Updated this week
- Supporting code for "LLMs for your iPhone: Whole-Tensor 4 Bit Quantization"☆11Mar 31, 2024Updated 2 years ago
- torchprime is a reference model implementation for PyTorch on TPU.☆49Mar 3, 2026Updated 7 months ago
- 在线图书借阅系统 - 2017 THU OOP课大作业☆14Jul 1, 2018Updated 8 years ago
- A library for efficient nucleotide sequence manipulation using 2-bit encoding☆12Sep 28, 2026Updated last month
- Tensor library for Zig☆12Nov 17, 2024Updated last year
- What's in a Prior? Learned Proximal Networks for Inverse Problems☆14Sep 11, 2025Updated last year
- ☆18Mar 18, 2024Updated 2 years ago
- ☆12Dec 28, 2023Updated 2 years ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Try to export the ONNX QDQ model that conforms to the AXERA NPU quantization specification. Currently, only w8a8 is supported.☆11Sep 10, 2024Updated 2 years ago
- Graph model execution API for Candle☆19Jul 27, 2025Updated last year
- AutodiffEngine☆13Apr 1, 2019Updated 7 years ago
- A fast, simple, multi-threaded string interning library.☆19Jul 11, 2025Updated last year
- antkillerfarm's crazy magic☆18Oct 3, 2024Updated 2 years ago
- ☆27Sep 10, 2026Updated last month
- GPU Performance and Cost Cheatsheet☆16Jul 13, 2025Updated last year