Predictive Intelligence Lab

A JAX implementation for the physics-informed learning of feedback linearizing representations.

Partial code base for "ADAPTIVE TOKENIZATION FOR VISION TRANSFORMER PDE SIMULATION" workshop submission.

A minimal but powerful library for exploring Mixture Density Networks and other multimodal modeling approaches in JAX.

Repository for the paper "Composite Bayesian Optimization In Function Spaces Using NEON -- Neural Epistemic Operator Networks", available at https://www.nature.com/articles/s41598-024-79621-7

Repository for some of the experiments presented in the paper "Deep Learning Alternatives of the Kolmogorov Superposition Theorem", Spotlight Paper in ICLR 2025.