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

Code for node DP of relation learning

All graph/GNN papers accepted at NeurIPS 2024.

Mamba SSM architecture

Official code for SPE

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.

SignNet and BasisNet

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

Junction Tree Variational Autoencoder for Molecular Graph Generation (ICML 2018)