- update hydra config to make experiment overrides smoother - fix directory naming - update readme
4 lines
73 B
Python
4 lines
73 B
Python
import jax
|
|
|
|
jax.config.update("jax_default_matmul_precision", "highest")
|