Official repository for the "Big Transfer (BiT): General Visual Representation Learning" paper.
created at May 11, 2020, 9:34 p.m.
Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX
created at April 3, 2020, 6:52 a.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.
Fast and Easy Infinite Neural Networks in Python
created at April 8, 2019, 4:48 p.m.
Composable kernels for scikit-learn implemented in JAX.
created at March 10, 2020, 6:59 p.m.
Fourier Features Let Networks Learn High Frequency Functions in Low Dimensional Domains
created at June 16, 2020, 11:54 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.
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
created at Oct. 29, 2018, 1:56 p.m.