Ben Everman
Results
1
comments of
Ben Everman
I was able to repro using the code @menzHSE provided. If you pass `retain_graph=True` to the `eval()` method it suppresses the error, but it trains much slower, as expected. [Here](https://github.com/ml-explore/mlx/blob/dfdb284e163c6580dc00a3c78ed58052e3389121/mlx/transforms.cpp#L160-L164)...