Probabilistic programming with NumPy powered by JAX for autograd and JIT compilation to GPU/TPU/CPU.

The tiniest of Gaussian Process libraries

Jupyter notebooks and associated material for NASSCC 2023 ML workshop

Topology and Material Optimization for Latent Heat Thermal Energy Storage Devices

A Differentiable Topology Optimization Framework for Multiphysics Fluidic Problems

Computational Fluid Dynamics in JAX

Python implementation of Elastica, an open-source software for the simulation of assemblies of slender, one-dimensional structures using Cosserat Rod theory.

Fourier Modal Method with Jax (official maintained repo)

Fourier modal method with Jax

GPU accelerated decision optimization

Efficient Differentiable n-d PDE solvers in JAX.

All the handwritten notes 📝 and source code files 🖥️ used in my YouTube Videos on Machine Learning & Simulation (https://www.youtube.com/channel/UCh0P7KwJhuQ4vrzc3IRuw4Q)

Differentiable Fluid Dynamics Package

AMPLE - A Material Point Learning Environment

Electromagnetic simulation (RCWA) & optimization package in Python

Electromagnetic FDTD Simulations in JAX

An open-source, code-first Python toolkit for building, evaluating, and deploying sophisticated AI agents with flexibility and control.

Rescure useful code in TensorNetwork by google for further personal usage.

😎 curated list of open source photonics projects

An open source parallel C++ package for the material point method (MPM)

Differentiable Material Point Method

Faster eigendecompositions for jax

[SIGGRAPH 2021] DiffAqua: A Differentiable Computational Design Pipeline for Soft Underwater Swimmers with Shape Interpolation

Massively parallel rigidbody physics simulation on accelerator hardware.

FEniCSx finite element basis evaluation library

Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax