BlackSamorez/tensor_parallel

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

↗ Open on GitHub

Stars

654

Forks

42

Language

Python

License

MIT

Last push

3 years ago

Python python machine-learning deep-learning pytorch nlp natural-language-processing pytorch-transformers

No related intel yet

This repo has not appeared in any of the sources the radar tracks. The collector runs on a schedule — check back once it covers this repo.