Commits

Branch selector

 main

User selector

Google-ML-Automation

Datepicker

All time

Commit History

Commits on Sep 1, 2024

• Update XLA dependency to use revision

  Show description for ebd5948
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  ebd5948

  Copy full SHA for ebd5948

Commits on Aug 31, 2024

• Update XLA dependency to use revision

  Show description for b6aedbc
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  b6aedbc

  Copy full SHA for b6aedbc

Commits on Aug 30, 2024

• Update XLA dependency to use revision

  Show description for bd486fc
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  bd486fc

  Copy full SHA for bd486fc

• Remove CPU test variant.

  Show description for 2f3990d
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  2f3990d

  Copy full SHA for 2f3990d

• Add cost estimate for forward pass of flash attention.

  Show description for 4c218fb
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  4c218fb

  Copy full SHA for 4c218fb

• Disable scoped tracing in pipelining

  Show description for 6a1adc8
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  6a1adc8

  Copy full SHA for 6a1adc8

Commits on Aug 29, 2024

• Update XLA dependency to use revision

  Show description for 45099b8
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  45099b8

  Copy full SHA for 45099b8

Commits on Aug 28, 2024

• Update XLA dependency to use revision

  Show description for 26619e2
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  26619e2

  Copy full SHA for 26619e2

• Fixes to api_benchmark.py. Testcases always fail without these fixes.

  Show description for eaefabe
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  eaefabe

  Copy full SHA for eaefabe

Commits on Aug 27, 2024

• Update XLA dependency to use revision

  Show description for db9e44f
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  db9e44f

  Copy full SHA for db9e44f

Commits on Aug 26, 2024

• Update XLA dependency to use revision

  Show description for 57c0d59
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  57c0d59

  Copy full SHA for 57c0d59

• Small fix for the jax trace dumping path

  Show description for c33ce85
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  c33ce85

  Copy full SHA for c33ce85

Commits on Aug 24, 2024

• Update XLA dependency to use revision

  Show description for e914362
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  e914362

  Copy full SHA for e914362

Commits on Aug 23, 2024

• Fix pallas int4->int8 conversion

  Show description for a2a351f
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  a2a351f

  Copy full SHA for a2a351f

• Update XLA dependency to use revision

  Show description for 6a5ca0b
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  6a5ca0b

  Copy full SHA for 6a5ca0b

• Update XLA dependency to use revision

  Show description for 20d13ab
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  20d13ab

  Copy full SHA for 20d13ab

• Refactor hermetic CUDA flags and update --config=cuda to add CUDA dependencies both for bazel build and bazel test phases.

  Show description for 279977c
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  279977c

  Copy full SHA for 279977c

Commits on Aug 22, 2024

• Update XLA dependency to use revision

  Show description for b6f2840
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  b6f2840

  Copy full SHA for b6f2840

Commits on Aug 21, 2024

• Update XLA dependency to use revision

  Show description for 810a919
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  810a919

  Copy full SHA for 810a919

• Fix MSAN use-of-uninitialized-value failure in array_test

  Show description for 8c7e798
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  8c7e798

  Copy full SHA for 8c7e798

Commits on Aug 20, 2024

• Update XLA dependency to use revision

  Show description for 9b3c19c
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  9b3c19c

  Copy full SHA for 9b3c19c

• Reverts 66a3f87

  Show description for 9bebf57
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  9bebf57

  Copy full SHA for 9bebf57

• Fix empty mesh size and abstract_mesh

  Show description for 16eb13e
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  16eb13e

  Copy full SHA for 16eb13e

Commits on Aug 19, 2024

• Update XLA dependency to use revision

  Show description for 292161a
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  292161a

  Copy full SHA for 292161a

• [shard_map docs]: Fix doc typos

  Show description for 3e764f6
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  3e764f6

  Copy full SHA for 3e764f6

Commits on Aug 18, 2024

• Update XLA dependency to use revision

  Show description for 05792c9
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  05792c9

  Copy full SHA for 05792c9

Commits on Aug 17, 2024

• Update XLA dependency to use revision

  Show description for b957f8b
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  b957f8b

  Copy full SHA for b957f8b

Commits on Aug 16, 2024

• Update XLA dependency to use revision

  Show description for 8b831b8
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  8b831b8

  Copy full SHA for 8b831b8

• 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

• [Easy] Refactor ragged_dot transpose, combine ragged_to_dense

  Show description for 9785368
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  9785368

  Copy full SHA for 9785368

Commits on Aug 15, 2024

• Update XLA dependency to use revision

  Show description for 417fcd5
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  417fcd5

  Copy full SHA for 417fcd5

• Set Clang as the default compiler in the build script.

  Show description for a498c1e
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  a498c1e

  Copy full SHA for a498c1e

Commits on Aug 14, 2024

• Update XLA dependency to use revision

  Show description for 2737a73
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  2737a73

  Copy full SHA for 2737a73

• Update XLA dependency to use revision

  Show description for 85fb66a
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  85fb66a

  Copy full SHA for 85fb66a

• Introduce hermetic CUDA in Google ML projects.

  Show description for 599c13a
  

  Google-ML-Automation

  authored and

  jax authors

  committed
  599c13a

  Copy full SHA for 599c13a

Pagination

Previous

Next