nmatare

dspy

DSPy: The framework for programming—not prompting—foundation models

submitit

Python 3.8+ toolbox for submitting jobs to Slurm

tensordict

TensorDict is a pytorch dedicated tensor container.

megablocks-public

mamba

Mamba SSM architecture

ml_collections

ML Collections is a library of Python Collections designed for ML use cases.

ml-engineering

Machine Learning Engineering Open Book

JaxMARL

Multi-Agent Reinforcement Learning with JAX

reverb

Reverb is an efficient and easy-to-use data storage and transport system designed for machine learning research

awesome-model-based-RL

A curated list of awesome model based RL resources (continually updated)

rliable

[NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.

flashbax

⚡ Flashbax: Accelerated Replay Buffers in JAX

tree

tree is a library for working with nested data structures

Eureka

Official Repository for "Eureka: Human-Level Reward Design via Coding Large Language Models" (ICLR 2024)

mctx

Monte Carlo tree search in JAX

LightZero

[NeurIPS 2023 Spotlight] LightZero: A Unified Benchmark for Monte Carlo Tree Search in General Sequential Decision Scenarios

jumanji

🕹️ A diverse suite of scalable reinforcement learning environments in JAX

open_spiel

OpenSpiel is a collection of environments and algorithms for research in general reinforcement learning and search/planning in games.

alpa

Training and serving large-scale neural networks with auto parallelization.

robotics_transformer

sagemaker-tensorflow-serving-container

A TensorFlow Serving solution for use in SageMaker. This repo is now deprecated.

flight-sql-server-example

An example Flight SQL Server implementation - with DuckDB and SQLite back-ends.

go-iex

A Go library for accessing the IEX Developer API.

environment

Neural MMO - A Massively Multiagent Environment for Artificial Intelligence Research

envlogger

A tool for recording RL trajectories.

keras-core

A multi-backend implementation of the Keras API, with support for TensorFlow, JAX, and PyTorch.

Sophia

Effortless plugin and play Optimizer to cut model training costs by 50%. New optimizer that is 2x faster than Adam on LLMs.

memory-maze

Evaluating long-term memory of reinforcement learning algorithms

online-dt

Online Decision Transformer

Temporal-Latent-Bottleneck-TF

An unofficial implementation of the paper "Temporal Latent Bottleneck: Synthesis of Fast and Slow Processing Mechanisms in Sequence Learning"