Popular repositories Loading

1. JaxLinOp JaxLinOp Public archive

   Linear Operators in JAX.

   Python 12

2. MOGPJax MOGPJax Public

   Multi-output Gaussian processes in JAX.

   Python 10

3. JaxKern JaxKern Public archive

   Kernel functions in JAX.

   Python 9 4

4. jax-decision-making jax-decision-making Public

   Sequential decision-making in JAX.

   Python 7 1

5. JaxUtils JaxUtils Public archive

   Model training utilities in JAX.

   Python 4 2

6. mytree mytree Public archive

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

   Python 2 1