Repository navigation

#

tensor

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Python
89133
2 小时前

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

Jupyter Notebook
12435
1 年前

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

Python
12218
3 小时前

Burn is a new comprehensive dynamic Deep Learning Framework built using Rust with extreme flexibility, compute efficiency and portability as its primary goals.

Rust
10251
19 小时前
arogozhnikov/einops

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

Python
8855
2 个月前

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

C++
4788
6 个月前

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

Elixir
2741
16 天前

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

Python
2737
1 天前

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

Python
2722
1 年前

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

2537
6 个月前
Jupyter Notebook
2115
5 个月前

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

C++
1290
5 天前

common in-memory tensor structure

C++
979
9 天前