Juan Emmanuel Johnson's repositories
BayesNewton
Bayes-Newton—A Gaussian process library in JAX, with a unifying view of approximate Bayesian inference as variants of Newton's method.
da_toolz
Data Assimilation tools with xarray
GPfY
Gaussian processes with spherical harmonic features in JAX
ocn-tools
Ocean Utility tools for working with oceanbench.
jejjohnson
My personal README.
finitevolX
Finite Volume tools in JAX
oceanbench
OceanBench - SSH edition
research_notebook
My personal research notebook with notes, tutorials, and resources written in Jupyterbook.
functa4eo
Using Functa for generalized interpolators for EO data
knerf
Minimal neural fields with keras-core.
fieldx
Custom xarray-like datastructures for fields with JAX
spectraldiffx
Spectral Difference tools with JAX
mfourdvar
Modern ML-Based Variational Data Assimilation framework
FiniteDiffX
Finite difference tools in JAX
eqx-trainer
A trainer module for the equinox (JAX) framework.
eqx-nerf
Neural Fields with JAX (equinox)