Repository Index

Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.

• ray
• distributed
• parallel
• machine-learning

DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

• deep-learning
• pytorch
• gpu
• machine-learning

Making large AI models cheaper, faster and more accessible

• deep-learning
• hpc
• large-scale
• data-parallelism

ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator

• deep-learning
• onnx
• neural-networks
• machine-learning

A fast, distributed, high performance gradient boosting (GBT, GBDT, GBRT, GBM or MART) framework based on decision tree algorithms, used for ranking, classification and many other machine learning tasks.

• gbdt
• gbm
• machine-learning
• data-mining

20+ high-performance LLMs with recipes to pretrain, finetune and deploy at scale.

• ai
• artificial-intelligence
• deep-learning
• large-language-models

DeepEP: an efficient expert-parallel communication library

An Easy-to-use, Scalable and High-performance Agentic RL Framework based on Ray (PPO & DAPO & REINFORCE++ & VLM & TIS & vLLM & Ray & Async RL)

• transformers
• vllm
• large-language-models
• raylib

A repo for distributed training of language models with Reinforcement Learning via Human Feedback (RLHF)

• machine-learning
• pytorch
• reinforcement-learning

Massively Parallel Deep Reinforcement Learning. 🔥

• pytorch
• reinforcement-learning
• ppo
• sac

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, TensorOpera AI (https://TensorOpera.ai) is your generative AI platform at scale.

• federated-learning
• deep-learning
• distributed-training
• edge-ai

A high-performance distributed training framework for Reinforcement Learning

• reinforcement-learning
• large-scale
• parallelization

PyTorch extensions for high performance and large scale training.

Determined is an open-source machine learning platform that simplifies distributed training, hyperparameter tuning, experiment tracking, and resource management. Works with PyTorch and TensorFlow.

• deep-learning
• machine-learning
• ml-platform
• ml-infrastructure

Distributed Deep learning with Keras & Spark

• spark
• keras
• neural-networks
• deep-learning