Open yaugenst-flex opened 3 weeks ago
We need some plugins to convert objective functions defined using autograd (since that is what we use) to PyTorch and possibly JAX. Proposed structure:
tidy3d/plugins ├── pytorch/ └── jax/
For JAX, we can probably just use agjax
We need some plugins to convert objective functions defined using autograd (since that is what we use) to PyTorch and possibly JAX. Proposed structure:
For JAX, we can probably just use agjax