reppo/src/jaxrl/__init__.py
Axel Brunnbauer 86fd47b04e - fix pyproject
- update hydra config to make experiment overrides smoother
- fix directory naming
- update readme
2025-07-15 22:20:32 -07:00

4 lines
73 B
Python

import jax
jax.config.update("jax_default_matmul_precision", "highest")