Pinned

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 17 3

2. 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 40 6

3. 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.

   Jupyter Notebook 9

4. awesome-neural-sbi awesome-neural-sbi Public

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

   63 3

5. consistency-models consistency-models Public

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

   Jupyter Notebook 16 1

6. eqnn-jax eqnn-jax Public

   Annotated implementations of equivariant neural networks in Jax.

   Jupyter Notebook 7 1