Tensors and Dynamic neural networks in Python with strong GPU acceleration
PyTorch tutorials and fun projects including neural talk, neural style, ...
Open deep learning compiler stack for cpu, gpu and specialized accelerators
Flexible and powerful tensor operations for readable and reliable code (...
NumPy & SciPy for GPU
MegEngine 是一个快速、可拓展、易于使用且支持自动求导的深度学习框架
LibRec: A Leading Java Library for Recommender Systems, see
Mars is a tensor-based unified framework for large-scale data computatio...
C++ DataFrame for statistical, Financial, and ML analysis -- in modern C...
Inference Llama 2 in one file of pure 🔥
Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code.
2-2000x faster ML algos, 50% less memory usage, works on all hardware - ...
TensorLy: Tensor Learning in Python.
A fast, ergonomic and portable tensor library in Nim with a deep learnin...
The Tensor Algebra Compiler (taco) computes sparse tensor expressions on...