GPU & Compute

Accelerate

Accelerate is a tool that simplifies the process of training and using PyTorch models across multiple GPUs, TPUs, and mi...

Acme

Acme is an open-source framework that simplifies the process of building and training reinforcement learning agents, off...

Advisor

Advisor is an open-source implementation of Google's Vizier for automating the process of hyperparameter tuning in machi...

Agent Lightning

Agent Lightning is a powerful framework designed to train and optimize AI agents using reinforcement learning techniques...

Amazon SageMaker

Amazon SageMaker makes it easier to train and deploy machine learning models using Hugging Face Transformers, offering s...

Arcade

Arcade is a powerful yet easy-to-use Python library that allows developers to create engaging, high-quality games. It pr...

ART

ART is a reinforcement learning framework that enables the training of multi-step agents for real-world tasks. It suppor...

auto_ml

Automates the process of building machine learning models with support for NLP, XGBoost, CatBoost, LightGBM. Outputs pro...

Auto_ViML

Auto_ViML is a comprehensive and scalable Python AutoML toolkit that handles imbalanced datasets, ensembling, stacking, ...

AutoGluon

AutoGluon simplifies the process of building machine learning models by automating key steps in model selection and tuni...

AutoGluon Mitra Classifier

The AutoGluon Mitra Classifier is an automated machine learning tool designed to simplify the process of building and de...

AutoKeras

AutoKeras is an open-source library that automates the process of building and optimizing machine learning models, makin...

AutoNLP

AutoNLP simplifies the process of training and deploying advanced natural language processing models by automating the e...

AutoPyTorch

AutoPyTorch automates the process of finding optimal neural network architectures and hyperparameters, making deep learn...

AutoRAG

AutoRAG is an open-source AutoML tool that optimizes the Retrieval-Augmented Generation (RAG) process, from dataset gene...

AutoSKLearn

AutoSKLearn is an automated machine learning toolkit that serves as a drop-in replacement for scikit-learn estimators, s...

Backprop

Backprop makes it simple to use, fine-tune, and deploy state-of-the-art machine learning models. It streamlines the proc...

Banana

Banana allows developers to host their machine learning models for inference on serverless GPUs, making it easy to integ...

Beam

Beam allows developers to develop on serverless GPUs, deploy highly performant APIs, and rapidly prototype ML models wit...

BMTrain

BMTrain is an efficient training framework designed to handle large-scale models. It provides a robust solution for deve...

Bullet3

Bullet Physics SDK provides real-time collision detection and physics simulations suitable for a wide range of applicati...

Chips-n-Salsa

Chips-n-Salsa is a Java library focused on genetic algorithms, evolutionary computation, and stochastic local search. It...

CleanRL

CleanRL offers high-quality, single-file implementations of popular Deep Reinforcement Learning algorithms like PPO, DQN...

cli-columns

cli-columns is a tool that allows developers to format text into columnated lists, ensuring proper alignment and handlin...

Coach

Reinforcement Learning Coach by Intel® AI Lab enables developers and researchers to easily experiment with the latest re...

Colorama

Colorama enables colored terminal output on Windows and cross-platform compatibility. It is essential for developers who...

Colossal-AI

Colossal-AI is an open-source deep learning system designed for efficient large-scale model training and inference. It a...

cONNXr

cONNXr is an ONNX runtime written in pure C99, designed to run inference on machine learning models across various frame...

DataRobot

DataRobot is an AI platform that democratizes data science by automating the entire machine learning lifecycle, from dat...

DEAP

DEAP is a powerful and flexible library that enables the design of evolutionary algorithms in Python. It provides tools ...