tensor

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Burn is a next generation tensor library and Deep Learning Framework that doesn't compromise on flexibility, efficiency and portability.

Open Machine Learning Compiler Framework

PyTorch tutorials and fun projects including neural talk, neural style, poem writing, anime generation (《深度学习框架PyTorch：入门与实战》)

NumPy & SciPy for GPU

Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)

The AI search platform

On-device AI across mobile, embedded and edge for PyTorch

MegEngine 是一个快速、可拓展、易于使用且支持自动求导的深度学习框架

C++ DataFrame for statistical, financial, and ML analysis in modern C++

A list of awesome compiler projects and papers for tensor computation and deep learning.

Mars is a tensor-based unified framework for large-scale data computation which scales numpy, pandas, scikit-learn and Python functions.

2-2000x faster ML algos, 50% less memory usage, works on all hardware - new and old.

Deep learning in Rust, with shape checked tensors and neural networks

CPU-first, eager-execution tensor/autograd runtime and LLM inference engine written in pure Zig