A PyTorch library for all things Reinforcement Learning (RL) for Combinatorial Optimization (CO)

A Library for Advanced Neural PDE Solvers.

python partial dependence plot toolbox

Official implementation of Transformer Neural Processes

A PyTorch library for implementing flow matching algorithms, featuring continuous and discrete flow matching implementations. It includes practical examples for both text and image modalities.

Fast Bayesian optimization, quadrature, inference over arbitrary domain with GPU parallel acceleration

Really Fast End-to-End Jax RL Implementations

A curated list for awesome discrete diffusion models resources.

Evolution of Heuristics

ALICE (Automated Learning and Intelligence for Causation and Economics) is a Microsoft Research project aimed at applying Artificial Intelligence concepts to economic decision making. One of its go…

LLM4AD: A Platform for Algorithm Design with Large Language Model

[TMLR 2025 & ICLR 2025 DeLTa] Official Implementation of Design Editing for Offline Model-based Optimization 🧬 🤖

Official implementation of paper: SFT Memorizes, RL Generalizes: A Comparative Study of Foundation Model Post-training

Normalizing-flow enhanced sampling package for probabilistic inference in Jax

Examining how large language models (LLMs) perform across various synthetic regression tasks when given (input, output) examples in their context, without any parameter update

Accelerated Quality-Diversity

Advanced evolutionary computation library built directly on top of PyTorch, created at NNAISENSE.

Official implementation of ICLR'25 paper "Offline Model-Based Optimization by Learning to Rank"

Benchmarked implementations of Offline RL Algorithms.

🚀 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

Finetune Llama 4, DeepSeek-R1, Gemma 3 & Reasoning LLMs 2x faster with 70% less memory! 🦥

Evolution Strategies in JAX 🦎

TorchCFM: a Conditional Flow Matching library

Clean, minimal, accessible reproduction of DeepSeek R1-Zero

verl: Volcano Engine Reinforcement Learning for LLMs

"Deep Generative Modeling": Introductory Examples

Official implementation of ICLR'25 paper "Offline Model-Based Optimization by Learning to Rank"

[ICLR 2025] Official Implementation of ParetoFlow: Guided Flows in Multi-Objective Optimization🧬🧬🧬