Normalizing Flows using JAX
created at March 9, 2020, 12:38 a.m.
Functional models and algorithms for sparse signal processing
created at Dec. 22, 2020, 8:31 a.m.
Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX
created at April 3, 2020, 6:52 a.m.
A Python package of computer vision models for the Equinox ecosystem.
created at July 24, 2022, 10:02 p.m.
Reimplementation of the UniRep protein featurization model.
created at Jan. 21, 2020, 3:30 p.m.
LoRA for arbitrary JAX models and functions
created at April 21, 2023, 11:50 a.m.
Probabilistic Programming and Nested sampling in JAX
created at July 31, 2020, 6 p.m.
Unofficial JAX implementations of deep learning research papers
created at Jan. 8, 2022, 1:40 p.m.
A differentiable cosmology library in JAX
created at May 1, 2020, 5:47 p.m.