Skip to content

Commit

Permalink
docs: Switch TensorFlow example for JAX (#89)
Browse files Browse the repository at this point in the history
* Demonstrate installing the 'jax' extra instead of 'tensorflow'.
  • Loading branch information
matthewfeickert authored Jan 7, 2025
1 parent 13bd4ee commit 166e7d3
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions book/introduction.md
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 166e7d3

Please # to comment.