smsharma
Physicist and machine learning researcher. IAIFI Fellow at MIT/Harvard. Probabilistic modeling/inference and AI for science/physics.
MIT, IAIFICambridge, MA
Pinned Repositories
AIrXiv
AI-powered arXiv research assistant prototype.
awesome-neural-sbi
Community-sourced list of papers and resources on neural simulation-based inference.
consistency-models
Implementation of Consistency Models (Song et al 2023) for few-step image generation in Jax.
eqnn-jax
Annotated implementations of equivariant graph neural networks in Jax: EGNN, SEGNN, NequIP.
hierarchical-inference
Hierarchical neural implicit inference over event ensembles. Code repository associated with https://arxiv.org/abs/2306.12584.
minified-generative-models
Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs, etc
mining-for-substructure-lens
Inference of substructure properties in strong lensing systems with machine learning. Code repository associated with https://arxiv.org/abs/1909.02005.
PAPERCLIP-Hubble
Semantic alignment of astronomical data with natural language using multi-modal models. (Jax) Code associated with https://arxiv.org/abs/2403.08851.
point-cloud-galaxy-diffusion
Transformer-guided diffusion for galaxy clustering. Code repository associated with https://arxiv.org/abs/2311.17141
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.
smsharma's Repositories
smsharma/AIrXiv
AI-powered arXiv research assistant prototype.
smsharma/awesome-neural-sbi
Community-sourced list of papers and resources on neural simulation-based inference.
smsharma/minified-generative-models
Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs, etc
smsharma/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.
smsharma/point-cloud-galaxy-diffusion
Transformer-guided diffusion for galaxy clustering. Code repository associated with https://arxiv.org/abs/2311.17141
smsharma/consistency-models
Implementation of Consistency Models (Song et al 2023) for few-step image generation in Jax.
smsharma/eqnn-jax
Annotated implementations of equivariant graph neural networks in Jax: EGNN, SEGNN, NequIP.
smsharma/hierarchical-inference
Hierarchical neural implicit inference over event ensembles. Code repository associated with https://arxiv.org/abs/2306.12584.
smsharma/lensing-neural-fields
Implicit neural representations for strong lensing source reconstruction. Code repository associated with https://arxiv.org/abs/2206.14820.
smsharma/jax-conditional-flows
Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.
smsharma/PAPERCLIP-Hubble
Semantic alignment of astronomical data with natural language using multi-modal models. (Jax) Code associated with https://arxiv.org/abs/2403.08851.
smsharma/deep-prob-prog-tutorials
Tutorials for the 2022 IAIFI Summer School, covering (deep) probabilistic programming with Jax and NumPyro.
smsharma/ChatJesseT
A cybernetic physicist sent back in time to fuse techniques from quantum field theory and machine learning. April Fools' 2023.
smsharma/eft-hod
Simulation-based priors connecting EFT of large-scale structure and halo occupation distribution (HOD) models.
smsharma/iaifi-hackathon-2024
Starter notebooks for self-supervised/multimodal machine learning and neural simulation-based inference, for the 2024 IAIFI Astro Hackathon.
smsharma/generative-spectra-lightcurves
p(spectra|light curves)
smsharma/jet-setting
Set-based generative modeling of particle jets using continuous normalizing flows.
smsharma/smsharma
smsharma/CV
LaTeX CV.
smsharma/iaifi.github.io
smsharma/neural-conservation-laws
Some quick checks from the paper Neural Conservation Laws: A Divergence-Free Perspective (NeurIPS 2022).
smsharma/smsharma.github.io
Source for my personal Jekyll-powered website, www.smsharma.io.
smsharma/cmb-differentiable-emu
Self-contained example of differentiable CMB parameter inference with power spectrum emulators.
smsharma/JaxKDTree
A Jax wrapper for cudaKDTree
smsharma/JeansGNN
Neural Simulation-based Inference with GNN for Jeans Modeling
smsharma/mlx-examples
Examples in the MLX framework
smsharma/pygsp
Graph Signal Processing in Python
smsharma/summer-school-2023
smsharma/symplectic-flows
Pedagogical normalizing flow that uses (reversible) symplectic integration.
smsharma/Zeus21
An effective model for the 21-cm signal at cosmic dawn