In the end, Zygote could not handle the code and I had to write my own adjoint. Maybe I will take a look at Zygote in the future.
Related topics
| Topic | Replies | Views | Activity | |
|---|---|---|---|---|
| Zygote.gradient(): Mutating arrays is not supported | 1 | 794 | August 18, 2020 | |
| Flux Zygote Gradient: Understanding Mutating arrays is not supported | 21 | 4212 | December 3, 2020 | |
| Is this impossible for Zygote Adjoint to handle | 4 | 613 | April 7, 2021 | |
| Mutation error in Zygote hessian | 2 | 201 | June 22, 2023 | |
| Zygote error - Mutating arrays is not supported | 12 | 1141 | October 26, 2022 |