Popular repositories Loading
Repositories
Showing 10 of 63 repositories
-
-
- flex_shard Public
Flexible parameter sharding and distributed optimization utilities for PyTorch training
- torchinfer Public
PyTorch inference enablement and optimization solution for serving DLRMv3 https//mlcommons.org/2026/02/dlrmv3-inference-meta/ model families
- autoparallel Public
An experimental implementation of compiler-driven automatic sharding of models across a given device mesh.
- tritonbench Public
Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
-
- 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.