Closed fehiepsi closed 3 months ago
Also fixing the enable_x64 logic which is not compatible with the latest version of jax. (we need to use a boolean instead of an integer for the setting)
enable_x64
Also fixing the
enable_x64
logic which is not compatible with the latest version of jax. (we need to use a boolean instead of an integer for the setting)