Pinned Loading

1. point-cloud-galaxy-diffusion point-cloud-galaxy-diffusion Public

   Transformer-guided diffusion for galaxy clustering. Code repository associated with https://arxiv.org/abs/2311.17141

   Jupyter Notebook 20 3

2. eqnn-jax eqnn-jax Public

   Annotated implementations of equivariant graph neural networks in Jax: EGNN, SEGNN, NequIP.

   Jupyter Notebook 17 1

3. minified-generative-models minified-generative-models Public

   Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs, etc

   Jupyter Notebook 117 8

4. PAPERCLIP-Hubble PAPERCLIP-Hubble Public

   Semantic alignment of astronomical data with natural language using multi-modal models. (Jax) Code associated with https://arxiv.org/abs/2403.08851 (COLM 2023).

   Jupyter Notebook 11

5. awesome-neural-sbi awesome-neural-sbi Public

   Community-sourced list of papers and resources on neural simulation-based inference.

   84 5

6. consistency-models consistency-models Public

   Implementation of Consistency Models (Song et al 2023) for few-step image generation in Jax.

   Jupyter Notebook 16 1