NEML2 updated to v3.1.0

Big updates

NEML2 now notionally supports any accelerator device available in pytorch. However, only "cpu" and "cuda" are under regular regression testing. "xpu" has been tested on a single machine. The others are unsupported, but may work.

pyzag package was updated to support new NEML2 interface.

Several new predictors added to improve convergence of complex models.

What's Changed

• tests/cpp: MSVC warning flags and short Inductor cache on Windows by @arsalanjameei in #426
• Make derived_output names user-overridable by default + surface them in neml2-syntax by @hugary1995 in #432
• AOTI substepping: ramp total-form forces, Newton step-norm capping, schema-configurable CP predictors by @hugary1995 in #433
• Capture solve-failure context on ConvergenceError (opt-in) by @hugary1995 in #434
• Fix eager call_batch_shape to strip parameter sub-batch axes by @hugary1995 in #437
• Fix stride-0 broadcast leak in sub_batch.expand_at by @hdt5kt in #443
• Port pyzag adapter to neml2 v3 by @hdt5kt in #419
• Add PerzynaPlasticFlowRateResidual: the Perzyna flow rule in inverted form by @hugary1995 in #444
• Fix AOTI export crash for custom predictors consuming a non-top-level given by @hugary1995 in #445
• Make the sub-batch declaration route-independent by @hugary1995 in #446
• Coordinate-descent predictor: 16 -> 5 on crystal plasticity, 15 -> 0 on viscoplasticity by @hugary1995 in #447
• Bump ruff from 0.15.21 to 0.16.5 by @dependabot[bot] in #455
• Make c10::LinAlgError recoverable at every NEML2 C++ boundary by @reverendbedford in #456
• Materialize broadcast initial guesses in Newton::solve_masked by @ChengHauYang in #458
• Bump pyright from 1.1.411 to 1.1.414 by @dependabot[bot] in #459
• Bump ruff from 0.16.5 to 0.16.7 by @dependabot[bot] in #460
• Generalize device handling to any torch accelerator (XPU, HIP, MPS) by @reverendbedford in #461
• Trivial change to support aurora by @reverendbedford in #462
• Bump ruff from 0.16.7 to 0.16.8 by @dependabot[bot] in #463
• Bump JamesIves/github-pages-deploy-action from 4.8.0 to 4.9.0 by @dependabot[bot] in #453
• Increment version number to 3.1.0 by @reverendbedford in #464

New Contributors

• @arsalanjameei made their first contribution in #426

Full Changelog: v3.0.7...v3.1.0