Uploaded image for project: 'AI Platform Core Components'
  1. AI Platform Core Components
  2. AIPCC-6916

Build jax from source instead of using pre-built wheel

XMLWordPrintable

    • Icon: Bug Bug
    • Resolution: Unresolved
    • Icon: Minor Minor
    • None
    • None
    • Accelerator Enablement
    • None
    • False
    • Hide

      None

      Show
      None
    • False

      .Description of problem:

          Currently, jax is used as a pre-built variant for tpu. The sdist files are available on pypi https://pypi.org/project/jax/0.6.1/#files, and can be built from source from pypi.

              rh-ee-jelevy Jenna Levy
              unassigned_jira Unassigned
              Frank's Team
              Votes:
              0 Vote for this issue
              Watchers:
              1 Start watching this issue

                Created:
                Updated: