TensorFlow

K

Keras

Community

Deep Learning for humans

S

scikit-learn

Community

scikit-learn: machine learning in Python

B

BudgetML

Community

Deploy a ML inference service on a budget in less than 10 lines of code.

D

Deepchecks

Community

Deepchecks: Tests for Continuous Validation of ML Models & Data. Deepchecks is a holistic open-source solution for all of your AI & ML validation needs, enabling to thoroughly test

D

Determined

Community

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

D

dstack

Community

Open framework for confidential AI

E

EasyFL

Community

An easy-to-use federated learning platform

E

EvalML

Community

EvalML is an AutoML library written in python.

F

FATE

Community

An Industrial Grade Federated Learning Framework

FA

Fiddler AI

Community

Fiddler Auditor is a tool to evaluate language models.

F

FLAML

Community

A fast library for AutoML and tuning. Join our Discord: https://discord.gg/Cppx2vSPVP.

F

Flower

Community

Flower: A Friendly Federated AI Framework

H

Horovod

Community

Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.

H

Hypernets

Community

A General Automated Machine Learning framework to simplify the development of End-to-end AutoML toolkits in specific domains.

K

Katib

Community

Automated Machine Learning on Kubernetes

KT

Keras Tuner

Community

A Hyperparameter Tuning Library for Keras

K

Keras

Community

Deep Learning for humans

K

Kserve

Community

Standardized Distributed Generative and Predictive AI Inference Platform for Scalable, Multi-Framework Deployment on Kubernetes

MS

Model Search

Community


M

Mosec

Community

A high-performance ML model serving framework, offers dynamic batching and CPU/GPU pipelines to fully exploit your compute machine

N

NNI

Community

An open source AutoML toolkit for automate machine learning lifecycle, including feature engineering, neural architecture search, model compression and hyper-parameter tuning.

P

PocketFlow

Community

An Automatic Model Compression (AutoMC) framework for developing smaller and faster AI applications.

S

Seldon-core

Community

An MLOps framework to package, deploy, monitor and manage thousands of production machine learning models

TF

TensorFlow Federated

Community

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

TM

TensorFlow Model Optimization

Community

A toolkit to optimize ML models for deployment for Keras and TensorFlow, including quantization and pruning.

T

TensorSpace

Community

Neural network 3D visualization framework, build interactive and intuitive model in browsers, support pre-trained deep learning models from TensorFlow, Keras, TensorFlow.js

TS

Triton Server (TRTIS)

Community

The Triton Inference Server provides an optimized cloud and edge inferencing solution.

V

Vegas

Community

AutoML tools chain

W

Wallaroo.AI

Community

Deploy, manage, optimize any model at scale across any environment from cloud to edge. Let's you go from python notebook to inferencing in minutes.

A

autokeras

Community

AutoML library for deep learning

BP

BERT: Pre-training of Deep Bidirectional Transformers for Language Understanding

Community

2018-10

D(

DEvol (DeepEvolution)

Community

Early POC of genetic neural architecture search

ET

Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformer

Community

Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformer

F

FedML

Community

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 runn

G

Gemma

Community

Checking your browser - reCAPTCHA

IL

Improving Language Understanding by Generative Pre-Training

Community

2018-06

KT

Keras Tuner

Community

A Hyperparameter Tuning Library for Keras

LM

Language Models are Unsupervised Multitask Learners

Community

2019-02

L

Ludwig

Community

Low-code framework for building custom LLMs, neural networks, and other AI models

MT

Mesh Tensorflow

Community

Mesh TensorFlow: Model Parallelism Made Easier

OG

Off Grid

Community

The Swiss Army Knife of Offline AI. Chat, Speak, and Generate Images - Privacy First, Zero Internet. Download an LLM and use it on your mobile device. No data ever leaves your phon

P

p-tuning-v2

Community

An optimized deep prompt tuning strategy comparable to fine-tuning across scales and tasks

TF

TensorFlow Federated

Community

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

TM

TensorFlow Model Optimization

Community

A toolkit to optimize ML models for deployment for Keras and TensorFlow, including quantization and pruning.

T

TFServing

Community

A flexible, high-performance serving system for machine learning models

V

Vegas

Community

AutoML tools chain

A

AIDE

Community

AIDE: AI-Driven Exploration in the Space of Code. The machine Learning engineering agent that automates AI R&D.

A

Aim

Community

Aim 💫 — An easy-to-use & supercharged open-source experiment tracker.

AT

Awesome Tensor Compilers

Community

A list of awesome compiler projects and papers for tensor computation and deep learning.

B

BentoML

Community

The easiest way to serve AI apps and models - Build Model Inference APIs, Job queues, LLM apps, Multi-model pipelines, and more!

C

ClearML

Community

ClearML - Auto-Magical CI/CD to streamline your AI workload. Experiment Management, Data Management, Pipeline, Orchestration, Scheduling & Serving in one MLOps/LLMOps solution

C

Comet

Community

Examples of Machine Learning code using Comet.ml

C

conda

Community

A system-level, binary package and environment manager running on all major operating systems and platforms.

D

Dragonfly

Community

An open source python library for scalable Bayesian optimisation.

D

DVC

Community

🦉 Data Versioning and ML Experiments

E

envd

Community

🏕️ Reproducible development environment for humans and agents

F

Feast

Community

The Open Source Feature Store for AI/ML

F

FeatherCNN

Community

FeatherCNN is a high performance inference engine for convolutional neural networks.

F

Featureform

Community

The Virtual Feature Store. Turn your existing data infrastructure into a feature store.

F

Flower

Community

Flower: A Friendly Federated AI Framework

F

Flyte

Community

Dynamic, resilient AI orchestration. Coordinate data, models, and compute as you build AI workflows.

F

Forward

Community

A library for high performance deep learning inference on NVIDIA GPUs.

GA

Guild AI

Community

Experiment tracking, ML developer tools

H

Harmonia

Community

Federated Learning Made Easy

H

Hopsworks

Community

Hopsworks - Data-Intensive AI platform with a Feature Store

H

HpBandSter

Community

a distributed Hyperband implementation on Steroids

H

Hyperband

Community

Tuning hyperparams fast with Hyperband

H

Hyperopt

Community

Distributed Asynchronous Hyperparameter Optimization in Python

H

hyperunity

Community

A toolset for black-box hyperparameter optimisation.

JN

Jupyter Notebooks

Community

Jupyter Interactive Notebook

KT

Keras Tuner

Community

A Hyperparameter Tuning Library for Keras

KP

Kubeflow Pipelines

Community

Machine Learning Pipelines for Kubeflow

K

Kubeflow

Community

Machine Learning Toolkit for Kubernetes

K

Kueue

Community

Kubernetes-native Job Queueing

L

LabNotebook

Community

LabNotebook is a tool that allows you to flexibly monitor, record, save, and query all your machine learning experiments.

M

Maniford

Community

A model-agnostic visual debugging tool for machine learning

M

Metaflow

Community

Build, Manage and Deploy AI/ML Systems

M

ModelDB

Community

Open Source ML Model Versioning, Metadata, and Experiment Management

M

ModelFox

Community

ModelFox makes it easy to train, deploy, and monitor machine learning models.

M

MOE

Community

A global, black box optimization engine for real world metric optimization.

N

NASGym

Community

A simple OpenAI Gym environment for Neural Architecture Search (NAS)

N

netron

Community

Visualizer for neural network, deep learning and machine learning models

O

ONNX-MLIR

Community

Representation and Reference Lowering of ONNX Models in MLIR Compiler Infrastructure

O

ormb

Community

Docker for Your ML/DL Models Based on OCI Artifacts

P

Pachyderm

Community

Data-Centric Pipelines and Data Versioning

P

PAI

Community

Resource scheduling and cluster management for AI

P

Polyaxon

Community

Open Source AI Infra & Engineering Control Plane

P

Prefect

Community

Prefect is a workflow orchestration framework for building resilient data pipelines in Python.

P

Primehub

Community

open-source MLOps platform

R

RoBO

Community

RoBO: a Robust Bayesian Optimization framework

S

Sacred

Community

Sacred is a tool to help you configure, organize, log and reproduce experiments developed at IDSIA.

S

scalene

Community

Scalene: a high-performance, high-precision CPU, GPU, and memory profiler for Python with AI-powered optimization proposals

S

Slurm

Community

Slurm: A Highly Scalable Workload Manager

S

Spearmint

Community

Spearmint Bayesian optimization codebase

S

Starwhale

Community

an MLOps/LLMOps platform

T

TensorBoard

Community

TensorFlow's Visualization Toolkit

TM

TensorFlow Model Optimization

Community

A toolkit to optimize ML models for deployment for Keras and TensorFlow, including quantization and pruning.

T

TFServing

Community

A flexible, high-performance serving system for machine learning models

T

tokenizers

Community

💥 Fast State-of-the-Art Tokenizers optimized for Research and Production

T

TVM

Community

Open Machine Learning Compiler Framework

V

Vald

Community

Vald. A Highly Scalable Distributed Vector Search Engine

V

Volcano

Community

A Cloud Native Batch System (Project under CNCF)

W&

Weights & Biases

Community

W&B Weave helps developers evaluate, monitor, and iterate continuously to deliver generative AI applications with confidence.

W

whylogs

Community

An open-source data logging library for machine learning models and data pipelines. 📚 Provides visibility into data quality & model performance over time. 🛡️ Supports privacy-pre

Y

Yunikorn

Community

Apache YuniKorn Core

Z

ZenML

Community

ZenML 🙏: One AI Platform from Pipelines to Agents. https://zenml.io.

Z

Zeno

Community

AI Data Management & Evaluation Platform

ZV

Zetane Viewer

Community

ML models and internal tensors 3D visualizer