Popular repositories Loading
Repositories
Showing 10 of 63 repositories
- tritonbench Public
Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
-
- torchinfer Public
PyTorch inference enablement and optimization solution for serving DLRMv3 https//mlcommons.org/2026/02/dlrmv3-inference-meta/ model families
- spmd_types Public
This module defines a type system for distributed training code, based off of JAX's sharding in types, but adapted for the PyTorch ecosystem.
-
- flex_shard Public
Flexible parameter sharding and distributed optimization utilities for PyTorch training
- compile-graph-break-site Public
This repository contains the source code for a static website that provides documentation for each "Graph Break" identified by a Graph Break ID (GBID).