Compatibility with latest JAX (fixed some warnings, fixed some crashes from the jax.ShapeDtypeStruct.weak_dtype breaking change).
Many solvers that need to calculate Jacobians will now preferentially use forward mode over reverse mode where possible. This generally improves speed. (#61)
Added warning that complex numbers are still a work in progress -- turns out that some facets of optimizing functions C->R autodifferentiably is perhaps still an open research question (?)
Dependabot will resolve any conflicts with this PR as long as you don't alter it yourself. You can also trigger a rebase manually by commenting @dependabot rebase.
Dependabot commands and options
You can trigger Dependabot actions by commenting on this PR:
- `@dependabot rebase` will rebase this PR
- `@dependabot recreate` will recreate this PR, overwriting any edits that have been made to it
- `@dependabot merge` will merge this PR after your CI passes on it
- `@dependabot squash and merge` will squash and merge this PR after your CI passes on it
- `@dependabot cancel merge` will cancel a previously requested merge and block automerging
- `@dependabot reopen` will reopen this PR if it is closed
- `@dependabot close` will close this PR and stop Dependabot recreating it. You can achieve the same result by closing it manually
- `@dependabot show ignore conditions` will show all of the ignore conditions of the specified dependency
- `@dependabot ignore this major version` will close this PR and stop Dependabot creating any more for this major version (unless you reopen the PR or upgrade to it yourself)
- `@dependabot ignore this minor version` will close this PR and stop Dependabot creating any more for this minor version (unless you reopen the PR or upgrade to it yourself)
- `@dependabot ignore this dependency` will close this PR and stop Dependabot creating any more for this dependency (unless you reopen the PR or upgrade to it yourself)
Updates the requirements on optimistix to permit the latest version.
Release notes
Sourced from optimistix's releases.
Commits
a8b7787
Update to latest release workflow + bump to Lineax compatible with latest JAXf320b67
version bump2c4066c
Version bump minimum Equinox version for compatibility with latest JAX.12b7fb8
Remove unused inner function58348db
Add py.type as per PEP 561b709f61
Update bisection.py docs1de4264
Allow the bisection root finder to expand the interval1c47169
Update mkdocs.ymlbfffa9b
Pass parameter value to optax's optimizer update4fc86e0
Switched from reverse mode to forward mode where possible.Dependabot will resolve any conflicts with this PR as long as you don't alter it yourself. You can also trigger a rebase manually by commenting
@dependabot rebase
.Dependabot commands and options
You can trigger Dependabot actions by commenting on this PR: - `@dependabot rebase` will rebase this PR - `@dependabot recreate` will recreate this PR, overwriting any edits that have been made to it - `@dependabot merge` will merge this PR after your CI passes on it - `@dependabot squash and merge` will squash and merge this PR after your CI passes on it - `@dependabot cancel merge` will cancel a previously requested merge and block automerging - `@dependabot reopen` will reopen this PR if it is closed - `@dependabot close` will close this PR and stop Dependabot recreating it. You can achieve the same result by closing it manually - `@dependabot show
📚 Documentation preview 📚: https://jaxspec--192.org.readthedocs.build/en/192/