PyTorch Labs

Pinned Repositories

applied-ai
Applied AI experiments and examples for PyTorch
Language:Python64 11 75
float8_experimental
This repository contains the experimental PyTorch native float8 training UX
Language:Python177 25 4218
gpt-fast
Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.
Language:Python5.3k 61 87476
segment-anything-fast
A batched offline inference oriented version of segment-anything
Language:Python1.1k 11 4065
superblock
A block oriented training approach for inference time optimization.
Language:Python151
torchfix
TorchFix - a linter for PyTorch-using code with autofix support
Language:Python62 8 1912
triton-cpu
An experimental CPU backend for Triton (https//github.com/openai/triton)
Language:C++164

PyTorch Labs's Repositories

pytorch-labs/gpt-fast
Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.
Language:Python5.3k 61 87476
pytorch-labs/segment-anything-fast
A batched offline inference oriented version of segment-anything
Language:Python1.1k 11 4065
pytorch-labs/float8_experimental
This repository contains the experimental PyTorch native float8 training UX
Language:Python177 25 4218
pytorch-labs/applied-ai
Applied AI experiments and examples for PyTorch
Language:Python64 11 75
pytorch-labs/torchfix
TorchFix - a linter for PyTorch-using code with autofix support
Language:Python62 8 1912
pytorch-labs/triton-cpu
An experimental CPU backend for Triton (https//github.com/openai/triton)
Language:C++164
pytorch-labs/superblock
A block oriented training approach for inference time optimization.
Language:Python151