Stars
A Curated List of Awesome Works in World Modeling, Aiming to Serve as a One-stop Resource for Researchers, Practitioners, and Enthusiasts Interested in World Modeling.
TORAX: Tokamak transport simulation in JAX
An RL-Gym for Challenge Problems in Data-Driven Modeling and Control of Fluid Dynamics.
Recommend new arxiv papers of your interest daily according to your Zotero libarary.
A Telegram bot to recommend arXiv papers
Pytorch implementation of the Variational Recurrent Neural Network (VRNN).
code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"
[ICLR 2025] CL-DiffPhyCon achieves closed-loop diffusion control of physical systems with significant speedup of sampling efficiency
Pytorch-based framework for solving parametric constrained optimization problems, physics-informed system identification, and parametric model predictive control.
PDEBench: An Extensive Benchmark for Scientific Machine Learning
Learning in infinite dimension with neural operators.
Latent Mamba Operator for Partial Differential Equations
All graph/GNN papers accepted at NeurIPS 2024.
OOD Dataset Curator and Benchmark for AI-aided Drug Discovery
This is an official implementation for "GRIT: Graph Inductive Biases in Transformers without Message Passing".
Simple numpy Implementation of paper Differentiable Ranks and Sorting using Optimal Transport.
Normalizing flows in PyTorch. Current intended use is education not production.
Simple illustrative examples for energy-based models in PyTorch
Pytorch implementations of density estimation algorithms: BNAF, Glow, MAF, RealNVP, planar flows
Repository for benchmarking graph neural networks (JMLR 2023)
Source code for our AAAI paper "Weisfeiler and Leman Go Neural: Higher-order Graph Neural Networks".
Paper list for equivariant neural network