Best practices for reporting an issue or bug relating to JAX on NVIDIA GPUs

When reporting a potential bug or issue, it really helps our engineers to get a comprehensive understanding of the environment and steps to reproduce the problem. This information will enable us to respond faster,

JAX developers respond to issues. We want to focus on fixing bugs and adding features.

  1. Provide details on the platforms you are using:
  • Linux distro and version
  • GPU type
  • NVIDIA driver version
  • CUDA version
  • cuDNN version
  • Python version [if using python]
  • JAX version
  1. Describe the issue

  2. Provide supporting code or data files