Differentiable convex optimization layers
created at Oct. 27, 2019, 10:42 p.m.
Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.
created at May 16, 2021, 1:34 p.m.
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
created at July 29, 2021, 2:21 a.m.
Unofficial JAX implementations of deep learning research papers
created at Jan. 8, 2022, 1:40 p.m.
Loopy belief propagation for factor graphs on discrete variables, in JAX!
created at June 9, 2021, 8:21 p.m.
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
created at Oct. 29, 2018, 1:56 p.m.