ROCm / pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration
219Updated this week

Related projects

Alternatives and complementary repositories for pytorch