Commits

Breadcrumbs

History for

onmain

User selector

All users

Datepicker

All time

Commit History

Commits on Dec 5, 2024

• [pallas] fix jumble test flakiness

  Show description for 84f3f99
  Google-ML-Automation

  committed
  84f3f99

  Copy full SHA for 84f3f99

Commits on Nov 8, 2024

• Skip flaky test on tpuv4

  Show description for 927d7fc
  Google-ML-Automation

  committed
  927d7fc

  Copy full SHA for 927d7fc

Commits on Oct 14, 2024

• Add a memory saving index rewrite step to vmap with ragged inputs over pallas_call.

  Show description for 1f0b572
  Google-ML-Automation

  committed
  1f0b572

  Copy full SHA for 1f0b572

Commits on Aug 20, 2024

• Reverts 66a3f87

  Show description for 9bebf57
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  9bebf57

  Copy full SHA for 9bebf57

Commits on Aug 19, 2024

• Rollback for: Implement initial vmap over pallas_call w/ ragged inputs (via jumbles)

  Show description for 66a3f87
  

  apaszke

  authored and

  jax authors

  committed
  66a3f87

  Copy full SHA for 66a3f87

Commits on Aug 16, 2024

• Implement initial vmap over pallas_call w/ ragged inputs (via jumbles)

  Show description for 24394a1
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  24394a1

  Copy full SHA for 24394a1

Loading