High-performance automatic differentiation of LLVM and MLIR.

This is a plugin for PyCBC to include waveform errors in parameter estimation.

This is the official repository for The Hundred-Page Language Models Book by Andriy Burkov

JAX-ReaxFF: A Gradient Based Framework for Extremely Fast Optimization of Reactive Force Fields

ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator

Codes to reproduce results in the paper

Investigation of biases on the parameters of GW sources in LISA and the Einstein Telescope from confusion noise of other resolved or unresolved signals.

From the LISA global fit to catalogs of single sources

Custom Bindings for Enzyme Automatic Differentiation Tool and Interfacing with JAX.

The Python ensemble sampling toolkit for affine-invariant MCMC

Short Fourier Transforms for Fresnel-weighted Template summation

Collecting research materials on neural samplers with diffusion/flow models

Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs

A multidimensional spline interpolation by Google/JAX

Using GPU to compute LISA response to gravitational waves in millisecond scale.

Fit and compare complex models reliably and rapidly. Advanced nested sampling.

Benchmark Blackjax samplers on Inference Gym models

Gravitational-wave specific proposals and reparameterisations for use with nessai

Combining gravitational wave harmonics for efficient evaluation of likelihood

Bayesian analysis of black hole ringdowns in gravitational-wave data.

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Normalizing Flows in JAX 🌊

Zotero is a free, easy-to-use tool to help you collect, organize, annotate, cite, and share your research sources.

Flow Annealed Importance Sampling Bootstrap (FAB) with JAX.

Zotero plugin, fetching publication metadata and cite counts (w/ and w/o self citations) from the INSPIRE-HEP database (and CrossRef cite counts for papers in other fields)

collection of favorite TikZ graphics

A beautiful, simple, clean, and responsive Jekyll theme for academics