Official repository for the "Big Transfer (BiT): General Visual Representation Learning" paper.
created at May 11, 2020, 9:34 p.m.
A differentiable cosmology library in JAX
created at May 1, 2020, 5:47 p.m.
Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX
created at April 3, 2020, 6:52 a.m.
Composable kernels for scikit-learn implemented in JAX.
created at March 10, 2020, 6:59 p.m.
Normalizing Flows using JAX
created at March 9, 2020, 12:38 a.m.
Reimplementation of the UniRep protein featurization model.
created at Jan. 21, 2020, 3:30 p.m.
Differentiable convex optimization layers
created at Oct. 27, 2019, 10:42 p.m.
Fast and Easy Infinite Neural Networks in Python
created at April 8, 2019, 4:48 p.m.