3 ms·indeed we are; but CUDA+Jax, which we need, isn't properly supported on pypiby insane_dreamer 2y agoindeed we are; but CUDA+Jax, which we need, isn't properly supported on pypi