Bugs in ML code, no matter how surprising, are almost always fixable once you create a reproducer than runs in 5 minutes or less. If your reproducer takes an hour or longer, for a confusing bug often better to work on golfing down its runtime rather than trying to fix the bug.