Agent skills for Obsidian

Beads - A memory upgrade for your coding agent

Parallelizing MCMC across the sequence length via Newton's method.

Normalizing flows in PyTorch

Google Authenticator in your terminal

probabilistic programming focused on fun

Finite State Machine implementations of MCMC algorithms in JAX for efficient vectorization.

Library for reading and processing ML training data.

Within-orbit Adaptive Leapfrog No-U-Turn Sampler

Bijections & normalizing flows with JAX/NNX

Functional near-infrared spectroscopy

A project to illustrate the power of Jax for simulating astrochemical kinetics.

Gaussian spectral decomposition with spatial coherence

Gaussian Process State Space Models in Python/JAX

Flexible, object-oriented implementation of ISM microphysics and chemistry with fast JAX solvers

JAX bindings to the Flatiron Institute Non-uniform Fast Fourier Transform (FINUFFT) library

differentiable (magneto)hydrodynamics for astrophysics in JAX

Transformer based foundation model for astronomy

Large-Scale Multimodal Dataset of Astronomical Data

Nonlinear optimisation (root-finding, least squares, ...) in JAX+Equinox. https://docs.kidger.site/optimistix/

Implementation of PSGD optimizer in JAX

Python interface to the CHIANTI atomic database

Run Stan models in the browser

Robust Heteroscedastic Matrix Factorisation in JAX

Custom plotting library based on my taste and requirements. Basically built on top of https://github.com/tcallister/makecorner

Frequently used code for generation publication-ready corner plots

Spectracles for stellar kinematics?

Bibliometric mapping of citations

A lightweight framework for building Physics-Informed Neural Networks (PINNs) with symbolic PDE definitions using SymPy and automatic differentiation in PyTorch. It provides flexible neural archite…

Companion repository for our paper applying spectrospatial models to LVM data.