jtamanas/optax

Optax is a gradient processing and optimization library for JAX.

Hey there superstar 🌟

As a member of both Stefano Profumo’s SCIPP Theory group at UC Santa Cruz and Joseph Hennawi’s ENIGMA group at UC Santa Barbara and Leiden Observatory, I work at the intersection of (high energy/astro)physics, deep learning, and statistics.

In my time as a graduate student, I’ve been lucky enough to work on projects with the world’s leading experts on applied deep learning. I have worked on deepening (pun intended) the understanding of current theories for a variety of topics: From beyond standard model particle physics, to the structure of the Milky Way, and the cosmological history of the universe.

You might know me from my work on:

1. Simulation-Based Inference: Approximate Sequential Bayesian algorithms written in JAX. Used with applications to high energy physics phenomenology and future quasar inference work.
2. Via Machinae: Unsupervised anomaly detection to discover stellar streams in the Milky Way.
3. Spectre: Approximate Bayesian algorithm for SOTA quasar continuum inference.