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

TranslatedPArallel Distributed Deep LEarning: Machine Learning Framework from Industrial Practice (Core framework of 'PaddlePaddle' (飞桨, Feijiang), for deep learning & machine learning: high-performance single-machine and distributed training, and cross-platform deployment)

• paddlepaddle
• deep-learning
• scalability
• 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

Easy-to-use and powerful LLM and SLM library with awesome model zoo.

• nlp
• embedding
• bert
• ernie

🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (including fp8), and easy-to-configure FSDP and DeepSpeed support

OneFlow is a deep learning framework designed to be user-friendly, scalable and efficient.

• deep-learning
• machine-learning
• deep-neural-networks
• ml

A unified ecosystem and workspace for large-scale federated learning of Large Models & LLMS. Store compute as Wifi

• compute-orchestration
• distributed-systems
• edge-computing
• edge-intelligence

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch

Flower: A Friendly Federated AI Framework

• flower
• federated-learning
• federated-learning-framework
• federated-analytics

A unified ecosystem and workspace for large-scale federated learning, bridging the gap between centralized model evolution and decentralized edge intelligence.

Simple and Distributed Machine Learning Python Library porting ML algorithms for Spark

• spark
• pyspark
• azure
• scala

The central nervous system and backend orchestrator for Fractal. It handles multi-tenant task scheduling, real-time TFLOPs budgeting, and federated weight aggregation using FedAvg.

• backend
• fedavg
• federated-learning
• flask

A unified inference and post-training framework for accelerated video generation.

• diffusers
• diffusion-models
• video-generation
• distillation

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

• machine-learning
• pytorch
• reinforcement-learning

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 distributed deep learning platform

• deep-learning

A high-performance distributed training framework for Reinforcement Learning

• reinforcement-learning
• large-scale
• parallelization

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

An open-source framework for machine learning and other computations on decentralized data.

Distributed Deep learning with Keras & Spark

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

Distributed machine learning platform

Framework and Library for Distributed Online Machine Learning

• machine-learning
• c-plus-plus
• ml
• distributed