TPU22

ChainRulesTestUtils crashes with a custom regularization function on a Flux Dense layer

4

I have been trying to write a custom reverse rule to a simple regularization function on a Flux Dense layer, and evaluate it with ChainRulesTestUtils. The function _gradient_ from Zygote...