Code

Add Layout support to jax.jit.#72

Code

Merged

copybara-service[bot] merged 1 commit intomainjax-ml/oryx:mainfrom test_621747200jax-ml/oryx:test_621747200Copy head branch name to clipboardApr 6, 2024

+30-13

Commits

Commits on Apr 6, 2024

• Add Layout support to jax.jit.

  Show description for 2a1300d
  

  yashk2810

  authored and

  The oryx Authors

  committed
  2a1300d

  Copy full SHA for 2a1300d