Repositories
30FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
Alerting system for the pytorch org
Tensors and Dynamic neural networks in Python with strong GPU acceleration
PyTorch Single Controller
A PyTorch native platform for training generative AI models
A Native-PyTorch Library for LLM Fine-tuning
Collective communications library with various primitives for multi-machine training.
DeepSpeed is a deep learning optimization library that makes distributed training easy, efficient, and effective.
Example models using DeepSpeed
The LLVM Project is a collection of modular and reusable compiler and toolchain technologies.
Ahead of Time (AOT) Triton Math Library
Development repository for the Triton language and compiler
No description provided.
No description provided.
A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
Experimental and Intriguing Tools for ROCm
HIP : Convert CUDA to Portable C++ Code
A collection of scripts related to building/testing various deep learning libraries
This repository hosts code that supports the testing infrastructure for the main PyTorch repo. For example, this repo hosts the logic to track disabled tests and slow tests, as well as our continuation integration jobs HUD/dashboard.
Data manipulation and transformation for audio signal processing, powered by PyTorch
The website for PyTorch
Datasets, Transforms and Models specific to Computer Vision
Continuous builder and binary build scripts for pytorch
🤗Transformers: State-of-the-art Natural Language Processing for Pytorch and TensorFlow 2.0.
:gear: A GitHub Action to send an email to multiple recipients
A GitHub action to create a pull request for changes to your repository in the actions workspace
Jenkins job definitions for OSSCI
Detectron2 is FAIR's next-generation platform for object detection and segmentation.
An implementation of a deep learning recommendation model (DLRM)
Python HIPify Tool