Awesome Data Analysis
Section: Tools · PyTorch wrapper for high-performance AI research.
FreshScore 80
Entry
PyTorch Lightning
Appears in 4 awesome lists
PyTorch Lightning pretrains, finetunes and deploys AI models on multiple GPUs, TPUs with zero code changes.
Open github.comlightning-ai/pytorch-lightning
Found in these lists
Awesome Production Machine Learning
Section: Computation and Communication Optimisation · PyTorch Lightning pretrains, finetunes and deploys AI models on multiple GPUs, TPUs with zero code changes.
FreshScore 92
awesome-python
Section: Machine Learning Frameworks · Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.
FreshScore 81
Awesome Python
Section: Deep Learning · Deep learning framework to train, deploy, and ship AI products Lightning fast.
FreshScore 94
Often listed next to it
TensorFlow
How to use the Hexagon Delegate to speed up model inference on mobile and edge devices. Also see blog post Accelerating TensorFlow Lite on Qualcomm Hexagon DSPs.
In 23 listsDetails
PyTorch
(label: good first issue) PyTorch is an open source machine learning library based on the Torch library, used for applications such as computer vision and natural language processing.
In 16 listsDetails
transformers
(formerly known as pytorch-transformers and pytorch-pretrained-bert) provides state-of-the-art general-purpose architectures (BERT, GPT-2, RoBERTa, XLM, DistilBert, XLNet, CTRL...) for Natural Language Understanding (NLU) and Natural Language Generation (NLG) with over 32+ pretrained models in…
In 14 listsDetails
Colossal-AI
(from Hpcaitech) - A Unified Deep Learning System for Large-Scale Parallel Training (1D, 2D, 2.5D, 3D and sequence parallelism, and ZeRO protocol).
In 14 listsDetails
Ray
A fast and simple framework for building and running distributed applications. Ray is packaged with RLlib, a scalable reinforcement learning library, and Tune, a scalable hyperparameter tuning library. ray.io
In 13 listsDetails
XGBoost
Scalable, Portable and Distributed Gradient Boosting (GBDT, GBRT or GBM) Library, for Python, R, Java, Scala, C++ and more. Runs on single machine, Hadoop, Spark, Flink and DataFlow. [Apache2]
In 11 listsDetails
Dask
| Python | - Parallel computing with task scheduling in Python with a Pandas like API
In 11 listsDetails
TensorFlow-Slim
Official TensorFlow repository of state-of-the-art (SOTA) models and modeling solutions. Contains reference implementations for BERT, ResNet, Transformer, and many more with pre-trained weights and training scripts. Apache 2.0 licensed.
In 12 listsDetails