pytorch_geometric
adaptdl
pytorch_geometric | adaptdl | |
---|---|---|
1 | 4 | |
14,506 | 395 | |
- | 0.0% | |
9.8 | 0.0 | |
about 2 years ago | about 1 year ago | |
Python | Python | |
MIT License | Apache License 2.0 |
Stars - the number of stars that a project has on GitHub. Growth - month over month growth in stars.
Activity is a relative number indicating how actively a project is being developed. Recent commits have higher weight than older ones.
For example, an activity of 9.0 indicates that a project is amongst the top 10% of the most actively developed projects that we are tracking.
pytorch_geometric
-
Introduction to PyTorch
PyTorch Geometric
adaptdl
-
Introduction to PyTorch
AdaptDL
-
[Discussion] Open source scheduler and queuing system for model training/inferencing tasks?
check out https://github.com/petuum/adaptdl. It natively supports AWS/EKS (for the autoscaling feature), otherwise it can run anywhere on Kubernetes.
-
Reduce cost by 3x in the cloud and improve GPU usage in shared clusters with AdaptDL for PyTorch
AdaptDL monitors training job performance in real-time, and elastically re-scales resources (GPUs, compute instances) while jobs are running. For each training job, AdaptDL automatically tunes the batch size, learning rate, and gradient accumulation. In the cloud (e.g. AWS), AdaptDL can auto-scale the number of provisioned Spot Instances. We’ve seen shared-cluster training jobs at Petuum and our partners complete 2–3x faster on average, with 3x cheaper cost in AWS using Spot Instances!
-
How we were able to achieve hyper-parameter tuning (HPT) for deep learning workflows at 1.5x faster in our clusters and 3x cheaper on AWS
To tackle the problem of long and expensive HPT workflows, our team at Petuum collaborated with Microsoft to integrate AdaptDL with Neural Network Intelligence (NNI). AdaptDL is an open-source tool in the CASL (Composable, Automatic, and Scalable Learning) ecosystem. AdaptDL offers adaptive resource management for distributed clusters, and reduces the cost of deep learning workloads ranging from a few training/tuning trials to thousands. NNI from the Microsoft open-source community, is a toolkit for automatic machine learning (AutoML) and hyper-parameter tuning.
What are some alternatives?
pytorch_geometric - Graph Neural Network Library for PyTorch
HandyRL - HandyRL is a handy and simple framework based on Python and PyTorch for distributed reinforcement learning that is applicable to your own environments.
Pytorch - Tensors and Dynamic neural networks in Python with strong GPU acceleration
alpa - Training and serving large-scale neural networks with auto parallelization.
pytorch-lightning - The lightweight PyTorch wrapper for high-performance AI research. Scale your models, not the boilerplate. [Moved to: https://github.com/PyTorchLightning/pytorch-lightning]
FedML - FEDML - The unified and scalable ML library for large-scale distributed training, model serving, and federated learning. FEDML Launch, a cross-cloud scheduler, further enables running any AI jobs on any GPU cloud or on-premise cluster. Built on this library, FEDML Nexus AI (https://fedml.ai) is your generative AI platform at scale.
PyNeuraLogic - PyNeuraLogic lets you use Python to create Differentiable Logic Programs
determined - Determined is an open-source machine learning platform that simplifies distributed training, hyperparameter tuning, experiment tracking, and resource management. Works with PyTorch and TensorFlow.
tutorials - PyTorch tutorials.
GNNs-Recipe - 🟠 A study guide to learn about Graph Neural Networks (GNNs)
PyTorch-NLP - Basic Utilities for PyTorch Natural Language Processing (NLP)