Change the repository type filter
All
Repositories list
67 repositories
- On-device AI across mobile, embedded and edge for PyTorch
- Tensors and Dynamic neural networks in Python with strong GPU acceleration
TensorRT
Public- Collective communications library with various primitives for multi-machine training.
- FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
- PyTorch native quantization and sparsity for training and inference
benchmark
Publicrl
PublicA modular, primitive-first, python-first PyTorch library for Reinforcement Learning.ignite
PublicHigh-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.tutorials
Publicaudio
Publicxla
Public- Datasets, Transforms and Models specific to Computer Vision
- CPU INFOrmation library (x86/x86-64/ARM/ARM64, Linux/Windows/Android/macOS/iOS)
pytorch_sphinx_theme
Publictorchchat
Public archivetext
Public archiveModels, data loaders and abstractions for language processing, powered by PyTorch