Juan Emmanuel Johnson's repositories
research_notebook
My personal research notebook with notes, tutorials, and resources written in Jupyterbook.
oceanbench
OceanBench - SSH edition
finitevolX
Finite Volume tools in JAX
BayesNewton
Bayes-Newton—A Gaussian process library in JAX, with a unifying view of approximate Bayesian inference as variants of Newton's method.
eqx-trainer
A trainer module for the equinox (JAX) framework.
FiniteDiffX
Finite difference tools in JAX
jejjohnson
My personal README.
spectraldiffx
Spectral Difference tools with JAX