[Mooncake](https://github.com/chalk-lab/Mooncake.jl) is intended as a drop-in replacement for Zygote, and the Mooncake team is happy to address GPU-related issues.