| pytorch/pytorch |
74,794 |
|
3,341 |
8,272 |
about 2 years ago |
39 |
November 15, 2023 |
13,261 |
other |
Python |
| Tensors and Dynamic neural networks in Python with strong GPU acceleration |
| apache/tvm |
10,828 |
|
0 |
3 |
about 2 years ago |
1 |
February 23, 2021 |
834 |
apache-2.0 |
Python |
| Open deep learning compiler stack for cpu, gpu and specialized accelerators |
| cupy/cupy |
7,482 |
|
96 |
107 |
about 2 years ago |
138 |
December 07, 2023 |
551 |
mit |
Python |
| NumPy & SciPy for GPU |
| MegEngine/MegEngine |
4,672 |
|
0 |
0 |
about 2 years ago |
0 |
|
163 |
apache-2.0 |
C++ |
| MegEngine 是一个快速、可拓展、易于使用且支持自动求导的深度学习框架 |
| NVIDIA/cutlass |
3,776 |
|
0 |
1 |
about 2 years ago |
1 |
February 22, 2022 |
53 |
other |
C++ |
| CUDA Templates for Linear Algebra Subroutines |
| elixir-nx/nx |
2,397 |
|
0 |
42 |
about 2 years ago |
16 |
November 13, 2023 |
19 |
|
Elixir |
| Multi-dimensional arrays (tensors) and numerical definitions for Elixir |
| tensorflow/haskell |
1,540 |
|
6 |
0 |
over 2 years ago |
4 |
August 03, 2018 |
36 |
apache-2.0 |
Haskell |
| Haskell bindings for TensorFlow |
| coreylowman/dfdx |
1,502 |
|
0 |
0 |
over 2 years ago |
17 |
July 27, 2023 |
79 |
other |
Rust |
| Deep learning in Rust, with shape checked tensors and neural networks |
| danielhanchen/hyperlearn |
1,387 |
|
0 |
0 |
almost 4 years ago |
0 |
|
0 |
apache-2.0 |
Jupyter Notebook |
| 2-2000x faster ML algos, 50% less memory usage, works on all hardware - new and old. |
| dmlc/mshadow |
1,036 |
|
0 |
0 |
over 6 years ago |
0 |
|
58 |
other |
C++ |
| Matrix Shadow:Lightweight CPU/GPU Matrix and Tensor Template Library in C++/CUDA for (Deep) Machine Learning |