From 471c5d300b2f2bebebed36ffc81ad54be34fb2c7 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: 702886844 --- clu/profiler.py | 5 +++-- clu_synopsis.ipynb | 2 +- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/clu/profiler.py b/clu/profiler.py index f946f8a..86af11c 100644 --- a/clu/profiler.py +++ b/clu/profiler.py @@ -31,8 +31,9 @@ def start(logdir: str, options=None): if options is not None: raise NotImplementedError( "'options' not supported by clu.profiler.start(). Please file an issue " - "at https://github.com/google/jax/issues requesting profiler option " - "support if you need this feature.") + "at https://github.com/jax-ml/jax/issues requesting profiler option " + "support if you need this feature." + ) if logdir is None: raise ValueError("Must specify logdir where profile should be written!") jax.profiler.start_trace(logdir) diff --git a/clu_synopsis.ipynb b/clu_synopsis.ipynb index 97a9008..68d2ad1 100644 --- a/clu_synopsis.ipynb +++ b/clu_synopsis.ipynb @@ -16,7 +16,7 @@ "https://github.com/google/CommonLoopUtils\n", "\n", "This package is usually used with\n", - "[JAX](https://github.com/google/jax)\n", + "[JAX](https://github.com/jax-ml/jax)\n", "/\n", "[Flax](https://github.com/google/flax)\n", "ML projects, but it can also be used with other ML frameworks.\n",