4 projects
agjax
A jax wrapper for autograd-differentiable functions.
find-jcmwave
Dynamically link jcmwave from JCMROOT into your environment
jaxwell
Jaxwell is JAX + Maxwell: an iterative solver for solving the finite-difference frequency-domain Maxwell equations on NVIDIA GPUs
javiche
A JAX wrapper around ceviche to make interoperability easier. In the future it might make sense to update ceviche itself to use JAX internally.