3 ms·
Yes because the pypi cupy/cudann packages now work seamlessly with jax. Until not long ago we had to use the conda packages.
by insane_dreamer 2y ago
Yes because the pypi cupy/cudann packages now work seamlessly with jax. Until not long ago we had to use the conda packages.