A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.
This repository hosts code that supports the testing infrastructure for the PyTorch organization. For example, this repo hosts the logic to track disabled tests and slow tests, as well as our continuation integration jobs HUD/dashboard.
A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind.
最近更新: 2天前torch::deploy (multipy for non-torch uses) is a system that lets you get around the GIL problem by running multiple Python interpreters in a single...
最近更新: 2天前A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.
最近更新: 2天前High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.
最近更新: 2天前