MLIR-based partitioning system
☆198Jul 22, 2026Updated this week
Alternatives and similar repositories for shardy
Users that are interested in shardy are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Backward compatible ML compute opset inspired by HLO/MHLO☆674Updated this week
- Tenstorrent MLIR compiler☆294Updated this week
- Repo for AI Compiler team. The intended purpose of this repo is for implementation of a PJRT device.☆73Updated this week
- TPP experimentation on MLIR for linear algebra☆155Updated this week
- Shared Middle-Layer for Triton Compilation☆340Dec 5, 2025Updated 7 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.
- IREE's PyTorch Frontend, based on Torch Dynamo.☆109Updated this week
- A Python compiler design toolkit.☆569Updated this week
- The Torch-MLIR project aims to provide first class support from the PyTorch ecosystem to the MLIR ecosystem.☆1,872Updated this week
- JaxPP is a library for JAX that enables flexible MPMD pipeline parallelism for large-scale LLM training☆80Jun 18, 2026Updated last month
- A machine learning compiler for GPUs, CPUs, and ML accelerators☆4,411Updated this week
- Tokamax: A GPU and TPU kernel library.☆250Updated this week
- ☆421Feb 24, 2026Updated 4 months ago
- C/C++ frontend for MLIR. Also features polyhedral optimizations, parallel optimizations, and more!☆623Jun 19, 2025Updated last year
- An experimental implementation of compiler-driven automatic sharding of models across a given device mesh.☆89Updated this week
- 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.
- The TT-Forge ONNX is a graph compiler designed to optimize and transform computational graphs for deep learning models, enhancing their p…☆65Updated this week
- incubator repo for CUDA-TileIR backend☆149Jul 10, 2026Updated last week
- A retargetable MLIR-based machine learning compiler and runtime toolkit.☆3,853Updated this week
- triton for dsa☆68Jul 10, 2026Updated last week
- Minimal yet performant LLM examples in pure JAX☆268Jul 4, 2026Updated 2 weeks ago
- The missing pieces (as far as boilerplate reduction goes) of the upstream MLIR python bindings.☆118Mar 4, 2026Updated 4 months ago
- Distributed Compiler based on Triton for Parallel Systems☆1,495Updated this week
- CUDA Tile IR is an MLIR-based intermediate representation and compiler infrastructure for CUDA kernel optimization, focusing on tile-base…☆1,001Jul 6, 2026Updated 2 weeks ago
- Wave: Python Domain-Specific Language for High Performance Machine Learning☆58Jun 29, 2026Updated 3 weeks ago
- 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.
- Bridging polyhedral analysis tools to the MLIR framework☆119Sep 9, 2023Updated 2 years ago
- Tilus is a tile-level kernel programming language with explicit control over shared memory and registers.☆489Jul 5, 2026Updated 2 weeks ago
- Representation and Reference Lowering of ONNX Models in MLIR Compiler Infrastructure☆1,037Updated this week
- ☆17Updated this week
- Intel® Extension for MLIR. A staging ground for MLIR dialects and tools for Intel devices using the MLIR toolchain.☆153Updated this week
- Github mirror of trition-lang/triton repo.☆181Updated this week
- Development repository for the Triton-Linalg conversion☆221Feb 7, 2025Updated last year
- PyTorch distributed training acceleration framework☆56Aug 13, 2025Updated 11 months ago
- Mirage Persistent Kernel: Compiling LLMs into a MegaKernel☆2,384Updated this week
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Website for CS 265☆34Dec 27, 2024Updated last year
- Python interface for MLIR - the Multi-Level Intermediate Representation☆270Nov 28, 2024Updated last year
- ☆142Updated this week
- Shared Middle-Layer for Triton Compilation☆27Updated this week
- extensible collectives library in triton☆97Mar 31, 2025Updated last year
- AMD RAD's multi-GPU Triton-based framework for seamless multi-GPU programming☆193Updated this week
- Experiments and prototypes associated with IREE or MLIR☆56Aug 9, 2024Updated last year