FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/

Staging ground for pytorch/pytorch maintainers to collaborate on release notes for PyTorch releases. Houses the artifacts that chronicle each release.

HIP Memory Manager (ROCm-DS)

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Pytorch domain library for recommendation systems

rocm based mlp tiny network based on tiny-cuda-nn

COLMAP - Structure-from-Motion and Multi-View Stereo

Datasets, Transforms and Models specific to Computer Vision

The C++ Standard Library for your entire system.

Self-contained GPU kernel profiler for ROCm. Zero roctracer/rocprofiler-sdk dependency. HSA-only interception, SQLite trace output, Perfetto compatible.

CUDA accelerated rasterization of gaussian splatting

PyTorch3D is FAIR's library of reusable components for deep learning with 3D data

C++ extensions in PyTorch

A fast, distributed, high performance gradient boosting (GBT, GBDT, GBRT, GBM or MART) framework based on decision tree algorithms, used for ranking, classification and many other machine learning …

The website for PyTorch

torchcomms: a modern PyTorch communications API

MSLK (Meta Superintelligence Labs Kernels) is a collection of PyTorch GPU operator libraries that are designed and optimized for GenAI training and inference, such as FP8 row-wise quantization and …

PyTorch native quantization and sparsity for training and inference

An implementation of a deep learning recommendation model (DLRM)

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 continua…

Fast and memory-efficient exact attention