Change the repository type filter
All
Repositories list
67 repositories
pytorch
PublicTensors and Dynamic neural networks in Python with strong GPU accelerationFBGEMM
PublicFB (Facebook) + GEMM (General Matrix-Matrix Multiplication) -https://code.fb.com/ml-applications/fbgemm/- On-device AI across mobile, embedded and edge for PyTorch
- 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.
- PyTorch native quantization and sparsity for training and inference
TensorRT
Public- A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.
kineto
Publictensordict
Public- Datasets, Transforms and Models specific to Computer Vision
cppdocs
Publicignite
PublicHigh-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.benchmark
Publiccpuinfo
PublicCPU INFOrmation library (x86/x86-64/ARM/ARM64, Linux/Windows/Android/macOS/iOS)xla
Publicgloo
PublicCollective communications library with various primitives for multi-machine training.- Models, data loaders and abstractions for language processing, powered by PyTorch