← Back to packages

jax-gpu

stable config
xlings install jax-gpu@cuda12

Description

Config: JAX with CUDA GPU support

Metadata

Latest Version cuda12
License Apache-2.0
Maintainers Google DeepMind
Categories ml gpu python config
Keywords jax cuda gpu machine-learning
Architectures x86_64, arm64
XVM Managed No

Platforms & Versions

Platform Latest Available Versions
Linux cuda12 cuda12

Dependencies

  • ros2:cuda-env
  • python

Links

Homepage Repository Documentation