Official repository for the "Big Transfer (BiT): General Visual Representation Learning" paper.
created at May 11, 2020, 9:34 p.m.
Normalizing Flows using JAX
created at March 9, 2020, 12:38 a.m.
A differentiable cosmology library in JAX
created at May 1, 2020, 5:47 p.m.
A selection of neural network models ported from torchvision for JAX & Flax.
created at June 14, 2020, 4:34 p.m.
Composable kernels for scikit-learn implemented in JAX.
created at March 10, 2020, 6:59 p.m.
Probabilistic Programming and Nested sampling in JAX
created at July 31, 2020, 6 p.m.
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
created at Oct. 29, 2018, 1:56 p.m.