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…☆154Updated 3 months ago
- PyTorch RFCs (experimental)☆130Updated 6 months ago
- Torch Distributed Experimental☆115Updated 7 months ago
- A library to analyze PyTorch traces.☆342Updated this week
- TorchFix - a linter for PyTorch-using code with autofix support☆136Updated last month
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆254Updated this week
- Home for OctoML PyTorch Profiler☆107Updated last year
- ☆161Updated 8 months ago
- TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iteration time for training/research and sup…☆349Updated this week
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆313Updated this week
- This repository contains the experimental PyTorch native float8 training UX☆222Updated 7 months ago
- Applied AI experiments and examples for PyTorch☆243Updated this week
- NVIDIA Resiliency Extension is a python package for framework developers and users to implement fault-tolerant features. It improves the …☆96Updated 2 weeks ago
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆108Updated 2 weeks ago
- torch::deploy (multipy for non-torch uses) is a system that lets you get around the GIL problem by running multiple Python interpreters i…☆179Updated 3 months ago
- A tensor-aware point-to-point communication primitive for machine learning☆253Updated 2 years ago
- ☆184Updated 2 weeks ago
- ☆186Updated 2 weeks ago
- The Triton backend for the PyTorch TorchScript models.☆144Updated last week
- Easy and lightning fast training of 🤗 Transformers on Habana Gaudi processor (HPU)☆175Updated this week
- PyTorch/XLA integration with JetStream (https://github.com/google/JetStream) for LLM inference"☆53Updated last month
- extensible collectives library in triton☆83Updated 5 months ago
- Implementation of a Transformer, but completely in Triton☆259Updated 2 years ago
- ☆34Updated this week
- A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.☆778Updated this week
- ☆38Updated 3 months ago
- ☆73Updated 4 months ago
- PyTorch centric eager mode debugger☆46Updated 2 months ago
- A Python library transfers PyTorch tensors between CPU and NVMe☆107Updated 3 months ago
- ☆288Updated last week