Projects I worked on...
Latent Diffusion models in JAX for Weather Forecasting
Open ↗
Flax implementation of Deep Quantum VMC