Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Feb 3, 2025

• Merge pull request #25812 from Cjkkkk:segment_ids

  Show description for 7164c6b
  Google-ML-Automation

  committed
  7164c6b

  Copy full SHA for 7164c6b

• Update XLA dependency to use revision

  Show description for 49821e8
  Google-ML-Automation

  committed
  49821e8

  Copy full SHA for 49821e8

• Add FFI example demonstrating the use of XLA's FFI state.

  Show description for 28afd25
  

  dfm

  authored and

  Google-ML-Automation

  committed
  28afd25

  Copy full SHA for 28afd25

• fix lint

  Cjkkkk

  committed
  8c4d6d6

  Copy full SHA for 8c4d6d6

• Reject invalid None in jax.NamedSharding(spec=None).

  Show description for cb188a0
  

  pschuh

  authored and

  Google-ML-Automation

  committed
  cb188a0

  Copy full SHA for cb188a0

• [pallas_mgpu] For loops can have **non-ref** accumulators for carries.

  Show description for b48d15d
  

  cperivol

  authored and

  Google-ML-Automation

  committed
  b48d15d

  Copy full SHA for b48d15d

• Stop passing None into jax.NamedSharding in preparation for followup which bans passing None in (in favor of PartitionSpec()

  Show description for da97ee2
  

  pschuh

  authored and

  Google-ML-Automation

  committed
  da97ee2

  Copy full SHA for da97ee2

• Reverts 3f73758

  Show description for 58d8a97
  

  hawkinsp

  authored and

  Google-ML-Automation

  committed
  58d8a97

  Copy full SHA for 58d8a97

• Merge pull request #26243 from jakevdp:einsum-asarray

  Show description for 57fa372
  Google-ML-Automation

  committed
  57fa372

  Copy full SHA for 57fa372

Commits on Feb 2, 2025

• Update XLA dependency to use revision

  Show description for af84143
  Google-ML-Automation

  committed
  af84143

  Copy full SHA for af84143

• Merge pull request #26174 from skye:cpu_configs

  Show description for de48ce2
  Google-ML-Automation

  committed
  de48ce2

  Copy full SHA for de48ce2

Commits on Feb 1, 2025

• [pallas/pallas_mgpu] Discharging run_scoped should not be discharging the intermediates

  Show description for b23f8f4
  

  cperivol

  authored and

  Google-ML-Automation

  committed
  b23f8f4

  Copy full SHA for b23f8f4

• [pallas] Support DMA start partial discharge and run_scoped() does its own partial discharge.

  Show description for 8649132
  

  cperivol

  authored and

  Google-ML-Automation

  committed
  8649132

  Copy full SHA for 8649132

• Update XLA dependency to use revision

  Show description for eb04fcb
  Google-ML-Automation

  committed
  eb04fcb

  Copy full SHA for eb04fcb

• [Pallas TPU] Support jnp.take_along_axis for 32-bit vreg-sized vector.

  Show description for ed952c8
  

  bythew3i

  authored and

  Google-ML-Automation

  committed
  ed952c8

  Copy full SHA for ed952c8

• [Mosaic TPU] Support dynamic gather along axis 0 or 1 for 32-bit vreg-sized vector.

  Show description for d8b9211
  

  bythew3i

  authored and

  Google-ML-Automation

  committed
  d8b9211

  Copy full SHA for d8b9211

• Move ragged_all_to_all test under appropriate test file

  Show description for c4e1763
  

  ghpvnist

  authored and

  Google-ML-Automation

  committed
  c4e1763

  Copy full SHA for c4e1763

Commits on Jan 31, 2025

• Merge pull request #25766 from carlosgmartin:nn_initializers_variance_scaling_mode_fan_geo_avg

  Show description for 872e6c0
  Google-ML-Automation

  committed
  872e6c0

  Copy full SHA for 872e6c0

• Merge pull request #26249 from jakevdp:fix-sterling

  Show description for a9f4dd7
  Google-ML-Automation

  committed
  a9f4dd7

  Copy full SHA for a9f4dd7

• Add mode='fan_geo_avg' to nn.initializers.variance_scaling.

  carlosgmartin

  committed
  96d3447

  Copy full SHA for 96d3447

• Reverts bb95113

  Show description for 3b2410f
  

  emilyfertig

  authored and

  Google-ML-Automation

  committed
  3b2410f

  Copy full SHA for 3b2410f

• Merge pull request #26195 from vfdev-5:use-tsan-numpy-in-tsan-ci

  Show description for 3f73758
  Google-ML-Automation

  committed
  3f73758

  Copy full SHA for 3f73758

• Fix dtype issue in stirling approximation

  jakevdp

  committed
  216bd9a

  Copy full SHA for 216bd9a

• Merge pull request #26247 from hawkinsp:tsan2

  Show description for 0bf8462
  Google-ML-Automation

  committed
  0bf8462

  Copy full SHA for 0bf8462

• Add additional tsan suppressions for races.

  Show description for d4c6657
  hawkinsp

  committed
  d4c6657

  Copy full SHA for d4c6657

• Merge pull request #26238 from hawkinsp:coretest

  Show description for 0ef2ccf
  Google-ML-Automation

  committed
  0ef2ccf

  Copy full SHA for 0ef2ccf

• Merge pull request #26239 from jakevdp:rotation-test

  Show description for b039f97
  Google-ML-Automation

  committed
  b039f97

  Copy full SHA for b039f97

• Merge pull request #26245 from jakevdp:fix-line

  Show description for 70aed64
  Google-ML-Automation

  committed
  70aed64

  Copy full SHA for 70aed64

• Merge pull request #26194 from jax-ml:fix-dist-init-runtime-error

  Show description for cb79ff4
  Google-ML-Automation

  committed
  cb79ff4

  Copy full SHA for cb79ff4

• delete unnecessary line

  jakevdp

  committed
  0df7f18

  Copy full SHA for 0df7f18

• Merge pull request #26236 from adonath:sphinx-copybutton-no-prompt-copy

  Show description for def5fd5
  Google-ML-Automation

  committed
  def5fd5

  Copy full SHA for def5fd5

• [pjrt] Use the PjRtMemorySpace version of BufferFromHostLiteral

  Show description for 5cad02c
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  5cad02c

  Copy full SHA for 5cad02c

• Avoid call to asarray in jnp.einsum

  jakevdp

  committed
  4e30a08

  Copy full SHA for 4e30a08

• Lower np.ndarray to DenseElementsAttr instead of ArrayAttr.

  Show description for 20555f6
  

  ghpvnist

  authored and

  Google-ML-Automation

  committed
  20555f6

  Copy full SHA for 20555f6

• Adjust conf.py to remove prompt from sphinx copybutton

  adonath

  committed
  dd7c203

  Copy full SHA for dd7c203

Pagination

Previous

Next