Repository navigation

#

tensor

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Python
93684
1 分钟前

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

Rust
13016
1 天前

Open deep learning compiler stack for cpu, gpu and specialized accelerators

Python
12691
3 天前

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

Jupyter Notebook
12676
2 年前
arogozhnikov/einops

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

Python
9204
2 个月前

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

C++
4800
1 年前

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

Python
3276
3 小时前

Multi-dimensional arrays (tensors) and numerical definitions for Elixir

Elixir
2817
3 天前

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

Python
2745
2 年前

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

2657
1 年前
Jupyter Notebook
2294
1 年前

The Tensor Algebra Compiler (taco) computes sparse tensor expressions on CPUs and GPUs

C++
1326
6 个月前