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

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

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

Ongoing research training transformer models at scale

• large-language-models
• model-para
• transformers

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

• nlp
• embedding
• bert
• ernie

Go ahead and axolotl questions

• fine-tuning
• llm

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 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

An implementation of model parallel autoregressive transformers on GPUs, based on the Megatron and DeepSpeed libraries

• deepspeed-library
• gpt-3
• transformers
• language-model

Flower: A Friendly Federated AI Framework

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

A Flexible and Powerful Parameter Server for large-scale machine learning

• machine-learning
• parameter-server
• spark
• scala

Model parallel transformers in JAX and Haiku

A Neural Net Training Interface on TensorFlow, with focus on speed + flexibility

• tensorflow
• deep-learning
• reinforcement-learning
• neural-networks

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

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

TensorFlowOnSpark brings TensorFlow programs to Apache Spark clusters.

• tensorflow
• spark
• yahoo
• machine-learning

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.

Proof Of Concept of DEcentralised Machine Learning on top of the Golem (https://golem.network/) architecture

• golem
• decentralised-machine-learning
• federated-learning