optax version #1

Open

opened

on Oct 20, 2023

AttributeError: module 'jaxlib.xla_extension' has no attribute 'CpuDevice'
What is the version of jax chex optax here?

Metadata

Assignees

No one assigned

Labels

No labels

No labels

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests