Pinned Repositories
applied-ai
Applied AI experiments and examples for PyTorch
attention-gym
Helpful tools and examples for working with flex-attention
float8_experimental
This repository contains the experimental PyTorch native float8 training UX
gpt-fast
Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.
LeanRL
LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.
segment-anything-fast
A batched offline inference oriented version of segment-anything
torchfix
TorchFix - a linter for PyTorch-using code with autofix support
torchft
PyTorch per step fault tolerance (actively under development)
triton-cpu
An experimental CPU backend for Triton (https//github.com/openai/triton)
tritonbench
Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
PyTorch Labs's Repositories
pytorch-labs/gpt-fast
Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.
pytorch-labs/segment-anything-fast
A batched offline inference oriented version of segment-anything
pytorch-labs/attention-gym
Helpful tools and examples for working with flex-attention
pytorch-labs/LeanRL
LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.
pytorch-labs/float8_experimental
This repository contains the experimental PyTorch native float8 training UX
pytorch-labs/applied-ai
Applied AI experiments and examples for PyTorch
pytorch-labs/torchfix
TorchFix - a linter for PyTorch-using code with autofix support
pytorch-labs/tritonbench
Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
pytorch-labs/torchft
PyTorch per step fault tolerance (actively under development)
pytorch-labs/triton-cpu
An experimental CPU backend for Triton (https//github.com/openai/triton)
pytorch-labs/superblock
A block oriented training approach for inference time optimization.
pytorch-labs/tokenizers
C++ implementations for various tokenizers (sentencepiece, tiktoken etc).
pytorch-labs/torchvision-extra-decoders
An extension of `torchvision` for decoding AVIF and HEIC images.
pytorch-labs/FACTO
Framework for Algorithmic Correctness Testing of Operators
pytorch-labs/ethos-u-core-driver-mirror
This is a mirror of https//git.mlplatform.org/ml/ethos-u/ethos-u-core-driver.git/
pytorch-labs/test-infra
pytorch-labs/tosa_serialization_lib-mirror
This is a mirror of https//git.mlplatform.org/tosa/serialization_lib.git/