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

Build jaxlib 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, jaxlib is used as a pre-built variant for tpu. Though the sdist files are not available on pypi, the source code is available from within the jax repository https://github.com/jax-ml/jax/tree/main/jaxlib and can therefore be built from source on builder. 

              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: