pytorch

Results 62 repositories owned by pytorch

torchx

304
Stars
96
Forks
Watchers

TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iteration time for training/research and support for E2E production ML pipelines when you're ready.

functorch

1.4k
Stars
102
Forks
Watchers

functorch is JAX-like composable function transforms for PyTorch.

xla

2.3k
Stars
409
Forks
Watchers

Enabling PyTorch on XLA Devices (e.g. Google TPU)

captum

4.6k
Stars
467
Forks
Watchers

Model interpretability and understanding for PyTorch

accimage

318
Stars
41
Forks
Watchers

high performance image loading and augmenting routines mimicking PIL.Image interface

android-demo-app

1.4k
Stars
575
Forks
Watchers

PyTorch android examples of usage in applications

benchmark

791
Stars
247
Forks
Watchers

TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.

botorch

2.9k
Stars
361
Forks
Watchers

Bayesian optimization in PyTorch

builder

315
Stars
199
Forks
Watchers

Continuous builder and binary build scripts for pytorch

contrib

388
Stars
45
Forks
Watchers

Implementations of ideas from recent papers