PyTorch Extension Library of Optimized Unique Operation
MIT License
Make PyTorch models up to 40% faster! Thunder is a source to source compiler for PyTorch. It enab...
End-to-end recipes for optimizing diffusion models with torchao and diffusers (inference and FP8 ...
PyTorch extensions for fast R&D prototyping and Kaggle farming
TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iter...
Training neural networks in PyTorch
Code snippets created for the PyTorch discussion board
Pytorch Custom CUDA kernel for searchsorted