Renderer for the harmony response format to be used with gpt-oss

DeepEP: an efficient expert-parallel communication library

DeepGEMM: clean and efficient FP8 GEMM kernels with fine-grained scaling

Guided Text Generation

Inference code for Mistral and Mixtral hacked up into original Llama implementation

Tensors and Dynamic neural networks in Python with strong GPU acceleration

The official repository for the Anything But Wrappers: Llama Edition Hackameetup

Continuous builder and binary build scripts for pytorch

A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.

Datasets, Transforms and Models specific to Computer Vision

Dynolog is a telemetry daemon for performance monitoring and tracing. It exports metrics from different components in the system like the linux kernel, CPU, disks, Intel PT, GPUs etc. Dynolog also …

A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.

A light-weight library for adding fault tolerance to large-scale PyTorch distributed training workloads.

Kubernetes Operator, helm charts, and production scripts for large-scale AIStore deployments on Kubernetes.

The website for PyTorch

Woboq CodeBrowser

SECOND for KITTI/NuScenes object detection

Spatial Sparse Convolution in PyTorch

Example repository for custom C++/CUDA operators for TorchScript

Translate - a PyTorch Language Library

An end-to-end PyTorch framework for image and video classification

Jenkins job definitions for OSSCI

Open Neural Network Exchange

Caffe2 is a lightweight, modular, and scalable deep learning framework.

PyTorch development for onnx

pytorch implementation of fast-neural-style

Computation using data flow graphs for scalable machine learning

Caffe2 implementation of Open Neural Network Exchange (ONNX)