Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX
created at April 3, 2020, 6:52 a.m.
Add a tqdm progress bar to your JAX scans and loops.
created at Jan. 16, 2023, 4:54 p.m.
Functional models and algorithms for sparse signal processing
created at Dec. 22, 2020, 8:31 a.m.
Normalizing Flows using JAX
created at March 9, 2020, 12:38 a.m.
Loopy belief propagation for factor graphs on discrete variables, in JAX!
created at June 9, 2021, 8:21 p.m.
🐈 Automatic differentiable spectrum modeling of exoplanets/brown dwarfs using JAX, compatible with NumPyro and JAXopt
created at Dec. 12, 2020, 3:46 a.m.
GPU/TPU accelerated nonlinear least-squares curve fitting using JAX
created at Aug. 8, 2022, 5 p.m.
A selection of neural network models ported from torchvision for JAX & Flax.
created at June 14, 2020, 4:34 p.m.
JAX + Flax implementation of "Combinatorial Optimization with Physics-Inspired Graph Neural Networks" by Schuetz et al.
created at Jan. 4, 2023, 11:09 p.m.
Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
created at Dec. 21, 2022, 9:20 a.m.
Composable kernels for scikit-learn implemented in JAX.
created at March 10, 2020, 6:59 p.m.
How to use the Flax Linen API to build a convolutional neural network model and train it for image classification (using TensorFlow Datasets).
created at Dec. 24, 2020, 3 a.m.