XiaobingSuper

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

intel-extension-for-pytorch

A Python package for extending the official PyTorch that can easily obtain performance on Intel platform

Deep-learning-math-kernel-research