AstroGen
Generative Modeling for Redshift-conditioned galaxy image synthesis.
Projects
Research code, model implementations, and experiments across scientific machine learning.
Fleiadex
Latent Diffusion models in JAX for Weather Forecasting
PsiFlax
Flax implementation of Deep Quantum VMC
MiniAcc
Serving large video diffusion models from a single GPU. A measured study of weight-traffic reduction: AdaLN sidecar plus INT8 weight quantization, with optional 8-bit attention, reaching about 2.5x faster requests on MiniMax-H3 with paired quality analysis.