You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
The bug I had with AMP is documented here: #699
It's very possible that this bug also affects autogram.Engine.compute_gramian, and not just autojac (we use vmap quite similarly internally). So it's possible that the two issues are related. I'll try to fix that soon.
@ogencoglu do you remember what went wrong when you tried compute_gramian on automatic mixed precision? Could it be the same error?
Activity
PierreQuinton commentedon Dec 19, 2025
Thanks for the issue!
It is not clear to me what the problem/goal is. Could you provide us with a minimal example of what you want to do?
ValerianRey commentedon Apr 24, 2026
Had a bug a few days ago with autojac.jac related to bfloat16. It wasn't autogram but it still may be related. I'll try to post a simple reproducer.
ValerianRey commentedon May 22, 2026
The bug I had with AMP is documented here: #699
It's very possible that this bug also affects
autogram.Engine.compute_gramian, and not just autojac (we use vmap quite similarly internally). So it's possible that the two issues are related. I'll try to fix that soon.@ogencoglu do you remember what went wrong when you tried
compute_gramianon automatic mixed precision? Could it be the same error?ogencoglu commentedon May 22, 2026
Unfortunately, I don't remember.
ValerianRey commentedon Jul 30, 2026
Closing in favor of #699