From 166e7d3376b8ab58a3eb1520447337fdc76bc1e2 Mon Sep 17 00:00:00 2001 From: Matthew Feickert Date: Mon, 6 Jan 2025 21:24:56 -0700 Subject: [PATCH] docs: Switch TensorFlow example for JAX (#89) * Demonstrate installing the 'jax' extra instead of 'tensorflow'. --- book/introduction.md | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/book/introduction.md b/book/introduction.md index 2197286..9b2f7bc 100644 --- a/book/introduction.md +++ b/book/introduction.md @@ -133,12 +133,12 @@ If you're installing from PyPI, you can also install with some of the "extras" t The 'xmlio' extra additionally installs [`uproot`](https://github.com/scikit-hep/uproot4) to read `ROOT` files. ::: -:::{tab-item} Use PyTorch and Tensorflow +:::{tab-item} Use PyTorch and JAX ``` -(pyhf-tutorial) $ python -m pip install 'pyhf[torch,tensorflow]' +(pyhf-tutorial) $ python -m pip install 'pyhf[torch,jax]' ``` -The 'torch' extra installs [`pytorch`](https://pytorch.org/) and the 'tensorflow' extra installs [`tensorflow`](https://www.tensorflow.org/). +The 'torch' extra installs [`pytorch`](https://pytorch.org/) and the 'jax' extra installs [`jax`](https://jax.readthedocs.io/). ::: :::{tab-item} Using Minuit Optimization