Olli Lupton
Results
83
comments of
Olli Lupton
Rebased again to resolve new conflicts.
> There was a recent change that has an impact on this implementation too: > > https://github.com/jax-ml/jax/blob/09d903fc9a60726d5b4e57769a6bdbd424d44b08/jax_plugins/cuda/__init__.py#L123-L141 > > I think you need something similar with try-except to load either...
Done, sorry about that. Tried to fix the `mypy` linter failure too.