pytorch
★ 105,490pytorch
PyTorch is a Python package for GPU-accelerated tensor computation and deep neural networks built on a tape-based autograd system, extensible via NumPy and SciPy.
HYSEN LABS DIRECTORY
pytorch
Verified repositories, classifications and analysis from the pytorch organization.
18 curated open-source projects
pytorch
★ 18,006vision
Datasets, Transforms and Models specific to Computer Vision
pytorch
★ 5,904torchtitan
A PyTorch native platform for training generative AI models
pytorch
★ 5,178executorch
On-device AI across mobile, embedded and edge for PyTorch
pytorch
★ 4,804ignite
High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.
pytorch
★ 3,638rl
A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.
pytorch
★ 3,027ao
PyTorch native quantization for training and inference
pytorch
★ 3,020TensorRT
PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
pytorch
★ 2,967audio
Data manipulation and transformation for audio signal processing, powered by PyTorch
pytorch
★ 2,808xla
Enabling PyTorch on XLA Devices (e.g. Google TPU)
pytorch
★ 1,601FBGEMM
FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
pytorch
★ 1,466gloo
Collective communications library with various primitives for multi-machine training.
pytorch
★ 1,444hub
Submission to https://pytorch.org/hub/
pytorch
★ 1,204extension-cpp
C++ extensions in PyTorch
pytorch
★ 1,194cpuinfo
CPU INFOrmation library (x86/x86-64/ARM/ARM64, Linux/Windows/Android/macOS/iOS)
pytorch
★ 1,056benchmark
TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.
pytorch
★ 1,046tensordict
TensorDict is a pytorch dedicated tensor container.
pytorch
★ 984helion
A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.