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.
☆94Updated 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 6 months ago
- PyTorch RFCs (experimental)☆132Updated 3 weeks ago
- A library to analyze PyTorch traces.☆391Updated this week
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆337Updated this week
- TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iteration time for training/research and sup…☆367Updated this week
- This repository contains the experimental PyTorch native float8 training UX☆224Updated 10 months ago
- ☆168Updated last year
- Torch Distributed Experimental☆116Updated 10 months ago
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆273Updated 3 weeks ago
- extensible collectives library in triton☆86Updated 2 months ago
- A tensor-aware point-to-point communication primitive for machine learning☆258Updated 2 years ago
- NVIDIA Resiliency Extension is a python package for framework developers and users to implement fault-tolerant features. It improves the …☆179Updated 2 weeks ago
- Implementation of a Transformer, but completely in Triton☆268Updated 3 years ago
- 🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.☆204Updated this week
- Applied AI experiments and examples for PyTorch☆277Updated 3 weeks ago
- Home for OctoML PyTorch Profiler☆113Updated 2 years ago
- jax-triton contains integrations between JAX and OpenAI Triton☆400Updated 3 weeks ago
- TorchFix - a linter for PyTorch-using code with autofix support☆143Updated 4 months ago
- The Triton backend for the PyTorch TorchScript models.☆152Updated this week
- A user-friendly tool chain that enables the seamless execution of ONNX models using JAX as the backend.☆112Updated this week
- Fast low-bit matmul kernels in Triton☆322Updated this week
- ☆219Updated this week
- torch::deploy (multipy for non-torch uses) is a system that lets you get around the GIL problem by running multiple Python interpreters i…☆180Updated last week
- oneCCL Bindings for Pytorch*☆97Updated last month
- Collection of kernels written in Triton language☆128Updated 2 months ago
- MLIR-based partitioning system☆91Updated this week
- ☆186Updated 2 weeks ago
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆166Updated this week
- ☆317Updated this week
- Easy and lightning fast training of 🤗 Transformers on Habana Gaudi processor (HPU)☆188Updated this week