Understand why tests fail with `jax == 0.5.0` #98

Open

opened

on Jan 23, 2025

They pass with jax == 0.4.23 or with:

    jax.config.update("jax_debug_nans", False)
    jax.config.update("jax_debug_infs", False)

to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Assignees

No one assigned

Labels

No labels

No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests