Skip to content
View homerjed's full-sized avatar

Highlights

  • Pro

Block or report homerjed

Block user

Prevent this user from interacting with your repositories and sending you notifications. Learn more about blocking users.

You must be logged in to block users.

Please don't include any personal information such as legal names or email addresses. Maximum 100 characters, markdown supported. This note will be visible to only you.
Report abuse

Contact GitHub support about this user’s behavior. Learn more about reporting abuse.

Report abuse
homerjed/README.md

Hi there 👋

I like using generative models to extract information in Bayesian inference problems on fundamental physics in cosmology.

Right now i'm working on

  • a simulation-based inference (SBI) package in jax,
  • field-level inference pipelines using generative models.

I'm interested in generative models...

...transformer models & geometric deep learning in general...

...and statistical problems in general

I also teach MSc Physics students in the Physik x AI labs at LMU Physik where I write teaching material that delivers machine learning insights derived from problems in physics.

My goal is to show students cutting edge algorithms and statistical methods that they will not learn anywhere else.

Pinned Loading

  1. sgm sgm Public

    Score-based Diffusion models in JAX.

    Python

  2. set_transformer set_transformer Public

    Implementation of a Set Transformer in JAX from the paper 'Set Transformer: A Framework for Attention-based Permutation-Invariant Neural Networks' by Lee et al. (2019)

    Jupyter Notebook

  3. vdm vdm Public

    Implementation and extension of 'Variational Diffusion Models' (Kingma++21) in JAX and Equinox.

    Python

  4. vdae vdae Public

    Implementation of 'Variational Diffusion Auto-encoder: Latent Space Extraction from Pre-trained Diffusion Models' (Batzolis++23) in JAX.

    Jupyter Notebook

  5. cct cct Public

    Implementation of 'Escaping the Big Data Paradigm with Compact Transformers' (Hassani++21) in JAX.

    Jupyter Notebook

  6. simple_vision_transformer simple_vision_transformer Public

    Lightweight implementation of 'An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale' (Dosovitskiy++21) in JAX.

    Jupyter Notebook