pytorch / test-infraLinks
This repository hosts code that supports the testing infrastructure for the PyTorch organization. For example, this repo hosts the logic to track disabled tests and slow tests, as well as our continuation integration jobs HUD/dashboard.
☆92Updated this week
Alternatives and similar repositories for test-infra
Users that are interested in test-infra are comparing it to the libraries listed below
Sorting:
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…☆157Updated 5 months ago
- PyTorch RFCs (experimental)☆132Updated last week
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆329Updated this week
- This repository contains the experimental PyTorch native float8 training UX☆223Updated 10 months ago
- A library to analyze PyTorch traces.☆379Updated last week
- ☆167Updated 11 months ago
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆268Updated this week
- extensible collectives library in triton☆87Updated 2 months ago
- Torch Distributed Experimental☆117Updated 9 months ago
- NVIDIA Resiliency Extension is a python package for framework developers and users to implement fault-tolerant features. It improves the …☆169Updated last week
- TorchFix - a linter for PyTorch-using code with autofix support☆141Updated 3 months ago
- Home for OctoML PyTorch Profiler☆113Updated 2 years ago
- Implementation of a Transformer, but completely in Triton☆265Updated 3 years ago
- oneCCL Bindings for Pytorch*☆97Updated last month
- Collection of kernels written in Triton language☆125Updated last month
- ☆310Updated last week
- A tensor-aware point-to-point communication primitive for machine learning☆257Updated 2 years ago
- ☆138Updated 2 weeks ago
- ☆47Updated this week
- ☆186Updated last week
- Easy and lightning fast training of 🤗 Transformers on Habana Gaudi processor (HPU)☆186Updated this week
- MLIR-based partitioning system☆86Updated this week
- TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iteration time for training/research and sup…☆366Updated last week
- Load compute kernels from the Hub☆139Updated this week
- PyTorch centric eager mode debugger☆47Updated 5 months ago
- 🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.☆196Updated this week
- ☆79Updated 6 months ago
- jax-triton contains integrations between JAX and OpenAI Triton☆395Updated this week
- Applied AI experiments and examples for PyTorch☆271Updated this week
- MLPerf™ logging library☆36Updated last month