diff --git a/readme.md b/readme.md index 601ab3b..fbada15 100644 --- a/readme.md +++ b/readme.md @@ -51,6 +51,7 @@ This section contains libraries that are well-made and useful, but have not nece - [FedJAX](https://github.com/google/fedjax) - Federated learning in JAX, built on Optax and Haiku. - [Equivariant MLP](https://github.com/mfinzi/equivariant-MLP) - Construct equivariant neural network layers. - [jax-resnet](https://github.com/n2cholas/jax-resnet/) - Implementations and checkpoints for ResNet variants in Flax. + - [Parallax](https://github.com/srush/parallax) - Immutable Torch Modules for JAX. - [jax-unirep](https://github.com/ElArkk/jax-unirep) - Library implementing the [UniRep model](https://www.nature.com/articles/s41592-019-0598-1) for protein machine learning applications. - [jax-flows](https://github.com/ChrisWaites/jax-flows) - Normalizing flows in JAX. - [sklearn-jax-kernels](https://github.com/ExpectationMax/sklearn-jax-kernels) - `scikit-learn` kernel matrices using JAX.