Daniel Dodd

gpjax

Last released Nov 1, 2024

Gaussian processes in JAX.

mytree

Last released Mar 22, 2023

Module pytrees that cleanly handle parameter trainability and transformations for JAX models.

GPyTree

Last released Mar 9, 2023

JaxPyTree

Last released Mar 9, 2023

JaxModule

Last released Mar 9, 2023

jaxutils

Last released Jan 8, 2023

Utility functions for JaxGaussianProcesses

mogpjax

Last released Dec 16, 2022

A JaxLinOp library.

jaxlinop

Last released Dec 2, 2022

A JaxLinOp library.

jax-linear-operator

Last released Oct 31, 2022

A JAX linear operator library.

samppl

Last released Sep 21, 2022

A simple probabilistic programming langauage in JAX.

BOJax

Last released Jul 28, 2022

HighJax

Last released Jul 27, 2022

GPBOx

Last released Jul 27, 2022