https://github.com/TRAIS-Lab/dattri/pull/193#discussion_r2683833611
#193 (comment)