PR #8919 Make GPU CUDA plugin require JAX

Make GPU CUDA plugin require JAX #8919

tengyifei wants to merge 4 commits into master from yifeit/cuda-plugin

tengyifei force pushed from e83d9b5f to 1b6d9ac7 1 year ago

Make GPU CUDA plugin require JAX

47d6ea1e

tengyifei force pushed from 1b6d9ac7 to 47d6ea1e 360 days ago

actually add jax dependencies to the CUDA plugin

2e674866

tengyifei force pushed from 4f623f5b to 2e674866 360 days ago

Don't bundle jax wheels into the cuda plugin artifact

c2024cb8

Update cudnn to 9.8.0

9a11f5d8

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone