pytorch

Results 64 repositories owned by pytorch

test-infra

60
Stars
55
Forks
Watchers

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 continuatio...

torchdistx

109
Stars
30
Forks
Watchers

Torch Distributed Experimental

torchdynamo

944
Stars
124
Forks
Watchers

A Python-level JIT compiler designed to make unmodified PyTorch programs faster.

torcheval

177
Stars
41
Forks
Watchers

A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to facilitate metric computation in distributed training and tools fo...

torchsnapshot

125
Stars
37
Forks
Watchers

A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind.

tutorials

7.7k
Stars
3.9k
Forks
Watchers

PyTorch tutorials.

tvm

450
Stars
68
Forks
Watchers

TVM integration into PyTorch

workshops

172
Stars
78
Forks
Watchers

This is a repository for all workshop related materials.

executorch

1.6k
Stars
250
Forks
Watchers

On-device AI across mobile, embedded and edge for PyTorch

tensordict

574
Stars
46
Forks
Watchers

TensorDict is a pytorch dedicated tensor container.