NVIDIA / cutlass

CUDA Templates and Python DSLs for High-Performance Linear Algebra

ritchieng / the-incredible-pytorch

The Incredible PyTorch: a curated list of tutorials, papers, projects, communities and more relating to PyTorch.