How to install jax?#
Date |
Category |
|---|---|
2023-12-21 21:38:20 |
Inverse Design |
The adjoint plugin requires the jax module, which is Autograd and XLA, brought together for high-performance numerical computing. We recommend running pip install “tidy3d[jax]” to install jax.