High-Speed Snake Simulator

ARTEMIS (Adaptive mesh Refinement Time-domain ElectrodynaMIcs Solver) couples the Maxwell's equations implementation in WarpX with classical equations that describe quantum material behavior (such …

Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python ⚡

A Python package for reading variably structured text files at scale

[NeurIPS 2025] Official repository for "Energy Matching: Unifying Flow Matching and Energy-Based Models for Generative Modeling"

Multiple dispatch in Python

Efficient Differentiable n-d PDE solvers in JAX.

✨🔬 A flexible diffraction simulator for exploring and visualizing physical optics.

Minimal JAX implementation of k-nearest neighbors using a k-d tree.

The code for the paper "verse Design of 3D Nanophotonic Devices with Structural Integrity Using Auxiliary Thermal Solvers".

Official implementation of "Explainable Reinforcement Learning via Dynamic Mixture Policies"

Interpolation and function approximation with JAX

👥 轻松创建研究组或组织网站 Easily create a stunning Research Group, Team, or Business Website with no-code

Learning in infinite dimension with neural operators.

Provide Python access to the NVML library for GPU diagnostics

Synthetic Metropolis Homography Dataset for Multiple Homography Fitting

HOPE-F Dataset for Multiple Fundamental Matrix Fitting

A list of paper and resources regarding Neural Cellular Automata

Code for the ICLR 2025 Paper "QPM: Discrete Optimization for Globally Interpretable Image Classification"

Differentiable Finite Element Method with JAX

Calculation of the Finite Difference Time Domain method in parallel on a GPU

gdsfactory plugins

Electromagnetic FDTD Simulations in JAX

Code for the AAAI 2024 Paper "Q-SENN: Quantized Self-Explaining Neural Networks"

Automatic Functional Differentiation in JAX

Visualize, create, and operate on pytrees in the most intuitive way possible.

JAX Arrays for human consumption

1 million FPS multi-agent driving simulator