Bump jaxtyping from 0.2.21 to 0.2.36 in /training

[dependabot]

Bumps jaxtyping from 0.2.21 to 0.2.36.

Release notes

Sourced from jaxtyping's releases.

jaxtyping v0.2.36

Fixed previous release that broke Python 3.9 compatibility! (Thanks @​baluyotraf! #269, #270, #271)

See https://github.com/patrick-kidger/jaxtyping/releases/tag/v0.2.35 for the corresponding release notes.

New Contributors

• @​baluyotraf made their first contribution in patrick-kidger/jaxtyping#270

Full Changelog: patrick-kidger/jaxtyping@v0.2.35...v0.2.36

jaxtyping v0.2.35

• Dropped dependency on typeguard==2.13.3. We now vendor our own copy of this. This means that jaxtyping now has zero dependencies. Note that more recent versions of typeguard do still seem to be a bit buggy in their handling of jaxtyping annotations, but we're now doing everything we can on this one :) (#199, #266)

• jaxtyping no longer produces references cycles, which should improve performance in a edge cases, e.g. when decorating temporary local functions. (Thanks @​ojw28! #258, #259, #260)

• Improved compaibility with cloudpickle: this library has a minor bug that produced crashes in some edge cases used alongside jaxtyping -- we've now hopefully worked around such issues once and for all. (#261, #262)

• Improved compatibility with poetry: this library has a bug in how it interpreters ~= version constraints in a nonstandard manner. We now use >= constraints instead. (Thanks @​norpadon! #257)

• Improved compatibility with mypy: now adding some extra typehints to help it infer the type of a jaxtyped-decorator function. (#254, #255)

New Contributors

• @​norpadon made their first contribution in patrick-kidger/jaxtyping#257
• @​ojw28 made their first contribution in patrick-kidger/jaxtyping#259

Full Changelog: patrick-kidger/jaxtyping@v0.2.34...v0.2.35

jaxtyping v0.2.34

• Compatibility with ray -- this fixes crashes with the error message has no attribute 'index_variadic' (#198, #237)
• Compatibility with Python 3.12: fixed deprecation warnings about ast.Str. (#236, thanks @​phinate!)
• Error message improvements + doc improvements (#233, #240, thanks @​padix-key and @​jjyyxx!)

New Contributors

• @​padix-key made their first contribution in patrick-kidger/jaxtyping#233
• @​phinate made their first contribution in patrick-kidger/jaxtyping#236
• @​jjyyxx made their first contribution in patrick-kidger/jaxtyping#240

Full Changelog: patrick-kidger/jaxtyping@v0.2.33...v0.2.34

jaxtyping v0.2.33

• Compatibility with Python 3.10 when using Any as the array type.
• Compatibility with generic array types.
• Array typevars now respect __constraints__

Full Changelog: patrick-kidger/jaxtyping@v0.2.32...v0.2.33

jaxtyping v0.2.32

• The array type can now be either Any or a TypeVar. In both cases this means that anything is allowed at runtime. As usual, static type checkers will only look at the array part of an annotation, so that an annotation of the form Float[T, "foo bar"] (where T = TypeVar("T")) will be treated as just T by static type checkers. This allows for expressing array-type-polymorphism with static typechecking. Here's an example:

... (truncated)

Commits

• 4307e19 New release for 3.9 compatibility; now testing this as well
• 7acd73e Add typing_extensions to allow usage of ParamSpec in python < 3.10
• c8c3edc Update description to reflect that we're no longer a jax-only project.
• ec6f72e Fix workflows
• dd24052 bump version
• d28a86c Improve docs to mention all the variant arraylikes we now support
• 8823419 ruff
• 72becb4 Now vendoring typeguard v2.13.3
• a1a9480 Hopefully fixing deserialization issues once and for all.
• a831de6 Call typechecker from isolated frame (#260)
• Additional commits viewable in compare view

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 <dependency name> 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)