Siddharth Mishra-Sharma's repositories
awesome-neural-sbi
Community-sourced list of papers and resources on neural simulation-based inference.
sbi-lecture-mit
Notebook to go along with a lecture for the MIT course 8.16: Data Science in Physics on neural simulation-based inference.
minified-generative-models
Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs, etc
consistency-models
Implementation of Consistency Models (Song et al 2023) for few-step image generation in Jax.
point-cloud-galaxy-diffusion
Diffusion modeling for galaxy clustering. Code repository associated with https://arxiv.org/abs/2311.17141
hierarchical-inference
Hierarchical neural implicit inference over event ensembles. Code repository associated with https://arxiv.org/abs/2306.12584.
jax-conditional-flows
Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.
lensing-neural-fields
Implicit neural representations for strong lensing source reconstruction. Code repository associated with https://arxiv.org/abs/2206.14820.
PAPERCLIP-Hubble
Semantic alignment of astronomical data with natural language using multi-modal models. (Jax) Code associated with https://arxiv.org/abs/2403.08851.
deep-prob-prog-tutorials
Tutorials for the 2022 IAIFI Summer School, covering (deep) probabilistic programming with Jax and NumPyro.
ChatJesseT
A cybernetic physicist sent back in time to fuse techniques from quantum field theory and machine learning. April Fools' 2023.
iaifi-hackathon-2024
Starter notebooks for self-supervised/multimodal machine learning and neural simulation-based inference, for the 2024 IAIFI Astro Hackathon.
jet-setting
Set-based generative modeling of particle jets using continuous normalizing flows.
generative-spectra-lightcurves
p(spectra|light curves)
neural-conservation-laws
Some quick checks from the paper Neural Conservation Laws: A Divergence-Free Perspective (NeurIPS 2022).
smsharma.github.io
Source for my personal Jekyll-powered website, www.smsharma.io.
cmb-differentiable-emu
Self-contained example of differentiable CMB parameter inference with power spectrum emulators.
JaxKDTree
A Jax wrapper for cudaKDTree
JeansGNN
Neural Simulation-based Inference with GNN for Jeans Modeling
mlx-examples
Examples in the MLX framework
symplectic-flows
Pedagogical normalizing flow that uses (reversible) symplectic integration.
Zeus21
An effective model for the 21-cm signal at cosmic dawn