Change the repository type filter
All
Repositories list
78 repositories
- PyTorch native quantization and sparsity for training and inference
pytorch
PublicTensors and Dynamic neural networks in Python with strong GPU acceleration- On-device AI across mobile, embedded and edge for PyTorch
- FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
- Pytorch domain library for recommendation systems
- 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.
- A lightweight library for PyTorch training tools and utilities
rl
PublicA modular, primitive-first, python-first PyTorch library for Reinforcement Learning.TensorRT
Public- Datasets, Transforms and Models specific to Computer Vision
- Models, data loaders and abstractions for language processing, powered by PyTorch
data
Publicpytorch.github.io
Public- Serve, optimize and scale PyTorch models in production
botorch
Public