BlackSamorez/tensor_parallel

Automatically split your PyTorch models on multiple GPUs for training & inference

/ 100

Established

656 stars and 3,412 monthly downloads. Used by 1 other package. No commits in the last 6 months. Available on PyPI.

Stale 6m

Maintenance 0 / 25

Adoption 19 / 25

Maturity 25 / 25

Community 14 / 25

Stars

656

Forks

Language

Python

License

MIT

Category

Last pushed

Jan 02, 2024

Monthly downloads

3,412

Commits (30d)

Dependencies

Reverse dependents

Get this data via API

curl "https://pt-edge.onrender.com/api/v1/quality/ml-frameworks/BlackSamorez/tensor_parallel"

Open to everyone — 100 requests/day, no key needed. Get a free key for 1,000/day.

Related frameworks

pymc-devs/pytensor

PyTensor allows you to define, optimize, and efficiently evaluate mathematical expressions...

tensorly/tensorly

TensorLy: Tensor Learning in Python.

arogozhnikov/einops

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

tensorpack/tensorpack

A Neural Net Training Interface on TensorFlow, with focus on speed + flexibility

lava-nc/lava-dl

Deep Learning library for Lava