Marcus Chiam comments

Repositories
Issues
Comments

Results 32 comments of


                                            Marcus Chiam

Test failures: "RuntimeError: Attempted to set 4 devices, but 1 CPUs already available: ensure that `set_n_cpu_devices` is executed before any JAX operation."

How are you running the test? Is it with `pytest -nauto tests/jax_utils_test.py`? Are you running any JAX operations before running this test? I was able to re-create the error by...

Should Flax return FrozenDicts or regular dicts?

Closing after #3193 landed.