This issue tracks the breaking change events from the upstream dependencies that break this library
check_field_needs_grad throws an error due to this issue. Waiting for upstream fix.
check_field_needs_grad
Fixed in v0.6.3