Using JAX as an AD backend

I’m very interested in helping to implement some experimental trials. I found some discussions that might be relevant:
How do we do move parts of the autodiff stack to the GPU? · Issue #1639 · stan-dev/math
Parallel autodiff v2 - Developers - The Stan Forums
Parallel autodiff v3 - Developers - The Stan Forums

1 Like