the official implementation of AMR-Transformer: Enabling Efficient Long-range Interaction for Complex Neural Fluid Simulation

Physics-informed neural networks in JAX — multi-GPU training, advanced PINN algorithms, and 16 benchmark examples

A python implemented flow visualiation framework

Fast CPU/GPU-accelerated extended-precision arithmetic in Julia

PyFR

A fixed-size step for dynamics, a giant leap for SciML

Exploring practical possibilities of approximating functions with frames rather than with a basis

With Uno, finally take full control of your SQP/barrier solver for nonlinearly constrained optimization

A Dual-space Multilevel Kernel-splitting (DMK) Framework for Discrete and Continuous Convolution

A code formatter for Julia with rules set in stone.

Code for the paper "XTrace: Making the most of every sample in stochastic trace estimation"

Code for Hutch++: Optimal Stochastic Trace Estimation

Stochastic trace estimation using JAX

Julia Package for Trace Estimation of implicitly available matrix on a smooth function such as inverse and log determinant.

Model and solve optimal control problems in Julia, both on CPU and GPU.

Open Source Optimization of Dynamic Multidisciplinary Systems

Approximate non-singular kernel functions by sum-of-exponentials or sum-of-gaussians by VP sum and model reduction.

An implementation of the VPMR algorithm, approximating decaying kernels by sum of exponentials.

Julia package to compute weights and points for the classical Gauss quadrature rules

A trajectory optimization library for Matlab

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

A logging system for SciML

Learning in infinite dimension with neural operators.

A domain specific language (DSL) for probabilistic graphical models

Prototype for sparse automatic differentiation in JAX + Julia

Cross-architecture parallel algorithms for Julia's CPU and GPU backends. Targets multithreaded CPUs, and GPUs via Intel oneAPI, AMD ROCm, Apple Metal, Nvidia CUDA.