/pytorch-uvm

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Primary LanguageC++OtherNOASSERTION

Stargazers