From 8cd614c51266e8ccd4c535324bf77301119736dc Mon Sep 17 00:00:00 2001 From: Jake Harmon Date: Wed, 4 Dec 2024 15:46:04 -0800 Subject: [PATCH] Update references to JAX's GitHub repo JAX has moved from https://github.com/google/jax to https://github.com/jax-ml/jax PiperOrigin-RevId: 702886847 --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 046f6ad..feba90e 100644 --- a/README.md +++ b/README.md @@ -20,7 +20,7 @@ install it directly from the repository, run: SOFIMA uses optical flow regularized with an elastic mesh to establish maps between data in different coordinate systems. Both the [flow estimator](flow_field.py) -as well as the [mesh solver](mesh.py) are implemented in [JAX](https://github.com/google/jax) +as well as the [mesh solver](mesh.py) are implemented in [JAX](https://github.com/jax-ml/jax) and will automatically take advantage of GPU acceleration if the hardware if available. A core data structure used throughout the project is a *coordinate map* stored