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.