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 1 year ago

actually add jax dependencies to the CUDA plugin

2e674866

tengyifei force pushed from 4f623f5b to 2e674866 1 year 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