homerjed Goto Github PK

asteroids_dyn_corr

Dynamic 2D cross-correlation to track asteroids and record photometrics.

cct

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

cdc_work

Work done at the Cosmic Dawn Centre, Copenhagen in Summer 2019.

cosmic_density_fields

cosmo_prob_notebooks

Notebooks of code for use in the 'Probability of the Cosmos; Quantum, Classical and Bayesian' tutorials.

cvae_julia

Conditional Variational Auto-Encoder in Julia.

distrax_bijector

Contextual bijector in Jax / Distrax.

dm-haiku

JAX-based neural network library

equinox

Elegant easy-to-use neural networks in JAX. https://docs.kidger.site/equinox/

etsformer-pytorch

Implementation of ETSformer, state of the art time-series Transformer, in Pytorch

frequentist_matching_priors

Implementation of 'Matching Bayesian and frequentist coverage probabilities when using an approximate data covariance matrix' (Percival++21) in JAX.

gan_tutorial

Tutorial for Generative Adversarial Networks (GANs) in PyTorch.

grf_lab

haiku_gmm

Simple Gaussian mixture model (GMM) in JAX, Haiku and Distrax.

hbnn

Hierarchical Bayesian network ensembles in JAX.

homerjed

lab

maf

Masked Autoregressive Flow (MAF) in JAX and Equinox

mnist_graph_net

Graph neural network global classifier on MNIST.

pydelfi

Density Estimation Likelihood-Free Inference with neural density estimators and adaptive acquisition of simulations

pyknit

Generate designs for pictures that can be knitted in wool.

relativity_vpy

Simulations of concepts in special relativity using vPython (version 7).

sample_flow

Simple code to sample a posterior with a normalizing flow likelihood in JAX.

self_compressing_neural_networks

Implementation of 'Self-Compressing Neural Networks' (Cséfalvay and Imber 2023) in JAX.

set_transformer

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)

simple_jax_flow

Affine Normalizing Flow written in Jax

simple_vision_transformer

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

sisyphus

A Terrible Program Built To Punish Every Fibre Of The Computer.

vdae

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

vdm

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