Skip to content

Commit

Permalink
Add OTT-JAX
Browse files Browse the repository at this point in the history
  • Loading branch information
michalk8 committed Apr 13, 2023
1 parent 41a7ef2 commit bd03571
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions readme.md
Original file line number Diff line number Diff line change
Expand Up @@ -99,6 +99,7 @@ This section contains libraries that are well-made and useful, but have not nece
- [jax-tqdm](https://github.com/jeremiecoullon/jax-tqdm) - Add a tqdm progress bar to JAX scans and loops. <img src="https://img.shields.io/github/stars/jeremiecoullon/jax-tqdm?style=social" align="center">
- [safejax](https://github.com/alvarobartt/safejax) - Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`. <img src="https://img.shields.io/github/stars/alvarobartt/safejax?style=social" align="center">
- [Kernex](https://github.com/ASEM000/kernex) - Differentiable stencil decorators in JAX.
- [OTT-JAX](https://github.com/ott-jax/ott) - Optimal transport tools in JAX. <img src="https://img.shields.io/github/stars/ott-jax/ott?style=social" align="center">

<a name="models-and-projects" />

Expand Down

0 comments on commit bd03571

Please sign in to comment.