-
Stars10
-
Rank 1,775,647 (Top 36 %)
-
LanguageJupyter Notebook
-
License
Creative Commons ... -
Created about 1 year ago
-
Updated about 1 month ago
Reviews
There are no reviews yet. Be the first to send feedback to the community and the maintainers!
Repository Details
Annotated implementations of equivariant graph neural networks in Jax: EGNN, SEGNN, NequIP.
More Repositories
1
AIrXiv
AI-powered arXiv research assistant prototype.Python
66
2
awesome-neural-sbi
Community-sourced list of papers and resources on neural simulation-based inference.
63
3
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.Jupyter Notebook
42
4
minified-generative-models
Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs, etcJupyter Notebook
40
5
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.Jupyter Notebook
28
6
point-cloud-galaxy-diffusion
Transformer-guided diffusion for galaxy clustering. Code repository associated with https://arxiv.org/abs/2311.17141Jupyter Notebook
17
7
consistency-models
Implementation of Consistency Models (Song et al 2023) for few-step image generation in Jax.Jupyter Notebook
16
8
hierarchical-inference
Hierarchical neural implicit inference over event ensembles. Code repository associated with https://arxiv.org/abs/2306.12584.Jupyter Notebook
10
9
lensing-neural-fields
Implicit neural representations for strong lensing source reconstruction. Code repository associated with https://arxiv.org/abs/2206.14820.TeX
9
10
PAPERCLIP-Hubble
Semantic alignment of astronomical data with natural language using multi-modal models. (Jax) Code associated with https://arxiv.org/abs/2403.08851.Jupyter Notebook
9
11
jax-conditional-flows
Normalizing flow models allowing for a conditioning context, implemented using Jax, Flax, and Distrax.Jupyter Notebook
9
12
phd-thesis
Princeton Ph.D. dissertation, defended August 2018.TeX
6
13
fermi-gce-flows
Simulation-based inference for the Galactic Center gamma-ray excess. Code repository associated with https://arxiv.org/abs/2110.06931.Jupyter Notebook
6
14
deep-prob-prog-tutorials
Tutorials for the 2022 IAIFI Summer School, covering (deep) probabilistic programming with Jax and NumPyro.Jupyter Notebook
6
15
gamma-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.Jupyter Notebook
6
16
GPSphere
Basic Gaussian processes on a sphere. Written at the Gaia Sprint 2019 (KITP).Jupyter Notebook
5
17
ChatJesseT
A cybernetic physicist sent back in time to fuse techniques from quantum field theory and machine learning. April Fools' 2023.Jupyter Notebook
4
18
neural-global-astrometry
Neural likelihood-ratio estimation for global astrometric lensing. Code repository associated with https://arxiv.org/abs/2110.01620.Jupyter Notebook
4
19
iaifi-hackathon-2024
Starter notebooks for self-supervised/multimodal machine learning and neural simulation-based inference, for the 2024 IAIFI Astro Hackathon.Jupyter Notebook
4
20
eft-hod
Simulation-based priors connecting EFT of large-scale structure and halo occupation distribution (HOD) models.Jupyter Notebook
4
21
semi-visible-jets
Models and event generation for semi-visible jets. Code repository associated with the paper https://arxiv.org/abs/1707.05326.Python
3
22
dark-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.Jupyter Notebook
3
23
edges-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.Jupyter Notebook
3
24
jet-setting
Set-based generative modeling of particle jets using continuous normalizing flows.Jupyter Notebook
2
25
prob-prog-8.S50-tutorial
A tutorial introducing probabilistic programming with Pyro for the MIT IAP Course 8.S50: Computational Data Science in PhysicsJupyter Notebook
2
26
smsharma
2
27
generative-spectra-lightcurves
p(spectra|light curves)Jupyter Notebook
2
28
smsharma.github.io
Source for my personal Jekyll-powered website, www.smsharma.io.JavaScript
1
29
neural-conservation-laws
Some quick checks from the paper Neural Conservation Laws: A Divergence-Free Perspective (NeurIPS 2022).Jupyter Notebook
1
30
astrometry-lensing-correlations
Angular power spectra for astrometric weak lensing. Code repository associated with the paper https://arxiv.org/abs/2003.02264.Jupyter Notebook
1