Stars
BM-MAE: Multimodal Masked Autoencoder Pre-training for 3D MRI-based Brain Tumor Analysis with Missing Modalities
Human in the loop Reinforcement Learning suite
Unofficial Scalable-Softmax Is Superior for Attention
DiffuLab is designed to provide a simple and flexible way to train diffusion models while allowing full customization of its core components.
Official repository for the TMLR paper "Self-Improvement for Neural Combinatorial Optimization: Sample Without Replacement, but Improvement"
Introduction to deep learning using manim animations
Learning 2-opt Heuristics for the TSP via Deep Reinforcement Learning
Efficient modeling interface for mathematical optimization in Python
Official implementation of Time-Constrained Robust MDP, NeurIPS 2024
Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"
A community-maintained Python framework for creating mathematical animations.
Recent research papers about Foundation Models for Combinatorial Optimization
[MICCAI 2024] DRIM: Learning Disentangled Representations from Incomplete Multimodal Healthcare Data
Gym Electric Motor (GEM): An OpenAI Gym Environment for Electric Motors
LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.
Modern implementation of the hybrid genetic search (HGS) algorithm specialized to the capacitated vehicle routing problem (CVRP). This code also includes an additional neighborhood called SWAP*.
The AI Scientist: Towards Fully Automated Open-Ended Scientific Discovery 🧑🔬
Feature-Based Landscape Analysis of Continuous and Constrained Optimization Problems
This library provides a comprehensive suite of algorithms to solve the Travelling Salesman Problem (TSP), ranging from Exact Algorithms, Heuristics, Metaheuristics and Reinforcement Learning techni…
[ICML'24 FM-Wild Oral] RouteFinder: Towards Foundation Models for Vehicle Routing Problems
A Python platform to perform parallel computations of optimisation tasks (global and local) via the asynchronous generalized island model.
Unifying Variational Autoencoder (VAE) implementations in Pytorch (NeurIPS 2022)
[SIGGRAPH Asia 2024, Journal Track] ToonCrafter: Generative Cartoon Interpolation
🤗 LeRobot: Making AI for Robotics more accessible with end-to-end learning