Top 12 Python Tensor Projects

Pytorch

336 77,783 10.0 Python

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Project mention: My Favorite DevTools to Build AI/ML Applications! | dev.to | 2024-04-23

TensorFlow, developed by Google, and PyTorch, developed by Facebook, are two of the most popular frameworks for building and training complex machine learning models. TensorFlow is known for its flexibility and robust scalability, making it suitable for both research prototypes and production deployments. PyTorch is praised for its ease of use, simplicity, and dynamic computational graph that allows for more intuitive coding of complex AI models. Both frameworks support a wide range of AI models, from simple linear regression to complex deep neural networks.

tvm

15 11,156 9.9 Python

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

Project mention: Making AMD GPUs competitive for LLM inference | news.ycombinator.com | 2023-08-09

Yes, this is coming! Myself and others at OctoML and in the TVM community are actively working on multi-gpu support in the compiler and runtime. Here are some of the merged and active PRs on the multi-GPU (multi-device) roadmap:
Support in TVM’s graph IR (Relax) - https://github.com/apache/tvm/pull/15447

InfluxDB

www.influxdata.com sponsored

Power Real-Time Data Analytics at Scale. Get real-time insights from all types of time series data with InfluxDB. Ingest, query, and analyze billions of data points in real-time with unbounded cardinality.
einops

17 7,897 7.4 Python

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

Project mention: Einops: Flexible and powerful tensor operations for readable and reliable code | news.ycombinator.com | 2023-12-12

cupy

21 7,774 9.9 Python

NumPy & SciPy for GPU

Project mention: CuPy: NumPy and SciPy for GPU | news.ycombinator.com | 2023-11-28

mars

0 2,675 5.7 Python

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

1 849 5.5 Python

common in-memory tensor structure

Project mention: Beginner projects/resources to learn about ML Compilers | /r/Compilers | 2023-04-27

For tensor layout, this is the standard for all libraries: https://github.com/dmlc/dlpack/blob/main/include/dlpack/dlpack.h

opt_einsum

1 802 0.0 Python

⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.
WorkOS

workos.com sponsored

The modern identity platform for B2B SaaS. The APIs are flexible and easy-to-use, supporting authentication, user identity, and complex enterprise features like SSO and SCIM provisioning.
norse

6 611 6.5 Python

Deep learning with spiking neural networks (SNNs) in PyTorch.
rwkvstic

2 144 6.7 Python

Framework agnostic python runtime for RWKV models
multidim_indexing

1 7 5.3 Python

Batch multidimensional indexing for pytorch
tensortrax

1 7 6.0 Python

Math on (Hyper-Dual) Tensors with Trailing Axes
tdb

1 1 6.2 Python

tdb: Simple, Lightweight Tool for Streamlined PyTorch Tensor and CUDA Debugging.

Project mention: TDB: Simple Tensor and CUDA debugging tool | news.ycombinator.com | 2024-04-13

NOTE: The open source projects on this list are ordered by number of github stars. The number of mentions indicates repo mentiontions in the last 12 Months or since we started tracking (Dec 2020).