minsoo2018

Luckily I solved this issue by using `hk.nets.MLP`, but I still don't know why the issue was resolved. ```python class MyLSTM(hk.Module): def __init__(self, hidden_size, output_size): super().__init__() self.hidden_size = hidden_size self.output_size...

Thanks, @tristandeleu, @mblondel, and all JAXOPT developers! Your advice has increased my understanding of implicit differentiation. Unfortunately, setting `implicit_diff=False` does not resolve the current issue. My next attempt to solve...