Reimplementation of the UniRep protein featurization model.
created at Jan. 21, 2020, 3:30 p.m.
Probabilistic Programming and Nested sampling in JAX
created at July 31, 2020, 6 p.m.
A library for programmatically generating equivariant layers through constraint solving
created at Sept. 11, 2020, 5:40 p.m.
Differentiable convex optimization layers
created at Oct. 27, 2019, 10:42 p.m.
Functional models and algorithms for sparse signal processing
created at Dec. 22, 2020, 8:31 a.m.
Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.
created at May 16, 2021, 1:34 p.m.
JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.
created at Jan. 16, 2021, 1:26 a.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.
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
created at Oct. 29, 2018, 1:56 p.m.