pytorch / test-infra
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.
☆90Updated this week
Alternatives and similar repositories for test-infra:
Users that are interested in test-infra are comparing it to the libraries listed below
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…☆155Updated 3 months ago
- PyTorch RFCs (experimental)☆131Updated 6 months ago
- TorchFix - a linter for PyTorch-using code with autofix support☆136Updated last month
- Torch Distributed Experimental☆115Updated 7 months ago
- A library to analyze PyTorch traces.☆348Updated last week
- TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iteration time for training/research and sup…☆354Updated this week
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆255Updated last week
- ☆162Updated 9 months ago
- Easy and lightning fast training of 🤗 Transformers on Habana Gaudi processor (HPU)☆177Updated this week
- ☆137Updated this week
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆314Updated this week
- ☆184Updated last month
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆109Updated 3 weeks ago
- JAX-Toolbox☆289Updated this week
- ☆290Updated this week
- Implementation of a Transformer, but completely in Triton☆260Updated 2 years ago
- Blazing fast training of 🤗 Transformers on Graphcore IPUs☆84Updated last year
- extensible collectives library in triton☆84Updated 6 months ago
- jax-triton contains integrations between JAX and OpenAI Triton☆386Updated last week
- This repository contains the experimental PyTorch native float8 training UX☆222Updated 7 months ago
- NVIDIA Resiliency Extension is a python package for framework developers and users to implement fault-tolerant features. It improves the …☆102Updated this week
- A tensor-aware point-to-point communication primitive for machine learning☆256Updated 2 years ago
- The Triton backend for the ONNX Runtime.☆140Updated last week
- ☆38Updated 3 months ago
- Applied AI experiments and examples for PyTorch☆249Updated this week
- ☆191Updated this week
- The Triton backend for the PyTorch TorchScript models.☆144Updated last week
- Home for OctoML PyTorch Profiler☆108Updated last year
- Testing framework for Deep Learning models (Tensorflow and PyTorch) on Google Cloud hardware accelerators (TPU and GPU)☆64Updated 4 months ago
- PyTorch centric eager mode debugger☆46Updated 3 months ago