Dylan Esguerra

icon company University of Washington icon location Seattle icon location Research Scientist Engineer at UW Medicine, MS Statistical Science UCSC

Results 1 issues of Dylan Esguerra

os.environ['XLA_FLAGS'] = '--xla_cpu_use_thunk_runtime=false' no longer working with Jax 0.6.0

3 comment

Hello I had experienced some slowdown in runtime using diffrax with Jax 0.5.3 and was recommended using this flag ``` os.environ['XLA_FLAGS'] = '--xla_cpu_use_thunk_runtime=false' no longer working with Jax 0.6.0 ```...