Researcher at Anthropic
awesome-neural-sbi. Community-sourced list of papers and resources on neural simulation-based inference.
163minified-generative-models. Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs, etc
142AIrXiv. AI-powered arXiv research assistant prototype.
74sbi-lecture-mit. Notebook to go along with a lecture for the MIT course 8.16: Data Science in Physics on neural simulation-based inference.
61eqnn-jax. Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.
43mining-for-substructure-lens. Inference of substructure properties in strong lensing systems with machine learning. Code repository associated with https://arxiv.org/abs/1909.02005.
32point-cloud-galaxy-diffusion. Transformer-guided diffusion for galaxy clustering. Code repository associated with https://arxiv.org/abs/2311.17141
25jax-conditional-flows. Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.
20consistency-models. Implementation of Consistency Models (Song et al 2023) for few-step image generation in Jax.
19PAPERCLIP-Hubble. Semantic alignment of astronomical data with natural language using multi-modal models. (Jax) Code associated with https://arxiv.org/abs/2403.08851 (COLM 2023).
17hierarchical-inference. Hierarchical neural implicit inference over event ensembles. Code repository associated with https://arxiv.org/abs/2306.12584.
14lensing-neural-fields. Implicit neural representations for strong lensing source reconstruction. Code repository associated with https://arxiv.org/abs/2206.14820.
12eft-hod. Simulation-based priors connecting EFT of large-scale structure and halo occupation distribution (HOD) models.
7GPSphere. Basic Gaussian processes on a sphere. Written at the Gaia Sprint 2019 (KITP).
7deep-prob-prog-tutorials. Tutorials for the 2022 IAIFI Summer School, covering (deep) probabilistic programming with Jax and NumPyro.
6phd-thesis. Princeton Ph.D. dissertation, defended August 2018.
6gamma-gp. Proof-of-principle application of Gaussian process modeling to gamma-ray analyses. Code repository associated with the paper https://arxiv.org/abs/2010.10450.
6fermi-gce-flows. Simulation-based inference for the Galactic Center gamma-ray excess. Code repository associated with https://arxiv.org/abs/2110.06931.
6neural-global-astrometry. Neural likelihood-ratio estimation for global astrometric lensing. Code repository associated with https://arxiv.org/abs/2110.01620.
4dark-photons-perturbations. Dark photon conversions in our inhomogeneous Universe. Code repository associated with the papers https://arxiv.org/abs/2002.05165 and https://arxiv.org/abs/2004.06733.
4ChatJesseT. A cybernetic physicist sent back in time to fuse techniques from quantum field theory and machine learning. April Fools' 2023.
4iaifi-hackathon-2024. Starter notebooks for self-supervised/multimodal machine learning and neural simulation-based inference, for the 2024 IAIFI Astro Hackathon.
4prob-prog-8.S50-tutorial. A tutorial introducing probabilistic programming with Pyro for the MIT IAP Course 8.S50: Computational Data Science in Physics
3semi-visible-jets. Models and event generation for semi-visible jets. Code repository associated with the paper https://arxiv.org/abs/1707.05326.
3edges-endpoints-21cm. Studying the effects of resonant photon injection in 21-cm observations. Code repository associated with the paper https://arxiv.org/abs/2009.03899.
3generative-spectra-lightcurves. p(spectra|light curves)
2jet-setting. Set-based generative modeling of particle jets using continuous normalizing flows.
2smsharma.
2twentyone-global. Simplified framework for modeling the global 21-cm absorption signal, with a focus on studying the implications of non-standard 21-cm CMB temperature evolution.
1symplectic-flows. Pedagogical normalizing flow that uses (reversible) symplectic integration.
1smsharma.github.io. Source for my personal Jekyll-powered website, www.smsharma.io.
1NPTFit. The Non-Poissonian Template Fit in python/cython: http://nptfit.readthedocs.io/
1neural-conservation-laws. Some quick checks from the paper Neural Conservation Laws: A Divergence-Free Perspective (NeurIPS 2022).
1astrometry-lensing-correlations. Angular power spectra for astrometric weak lensing. Code repository associated with the paper https://arxiv.org/abs/2003.02264.
1iaifi.github.io. SCSS
1CV. LaTeX CV.
1