Commits

Branch selector

 a24420e

User selector

All users

Datepicker

All time

Commit History

Commits on Sep 30, 2024

• [Pallas TPU] Add lowering for lax.cos_p

  Show description for a24420e
  

  ayaka14732

  authored and

  Google-ML-Automation

  committed
  a24420e

  Copy full SHA for a24420e

• [Pallas TPU] Consolidate OpsExtraTest into OpsTest

  Show description for 23ce5a1
  

  ayaka14732

  authored and

  Google-ML-Automation

  committed
  23ce5a1

  Copy full SHA for 23ce5a1

• Merge pull request jax-ml#23995 from jakevdp:trapezoid-doc

  Show description for c557db0
  Google-ML-Automation

  committed
  c557db0

  Copy full SHA for c557db0

• [pallas:triton] Do not DCE the jaxpr in the lowering pass

  Show description for d74d3da
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  d74d3da

  Copy full SHA for d74d3da

• [Mosaic TPU] Force offset to 0 when inferring input has offset out of the first tile.

  Show description for 4a596ae
  

  bythew3i

  authored and

  Google-ML-Automation

  committed
  4a596ae

  Copy full SHA for 4a596ae

• Merge pull request jax-ml#24025 from jakevdp:gradient-doc

  Show description for cdc7278
  Google-ML-Automation

  committed
  cdc7278

  Copy full SHA for cdc7278

• Better docs for jnp.gradient

  Show description for 36d6bb9
  jakevdp

  committed
  36d6bb9

  Copy full SHA for 36d6bb9

• Merge pull request jax-ml#23505 from sergachev:fix_cudnn_fusion_test

  Show description for 3766f88
  Google-ML-Automation

  committed
  3766f88

  Copy full SHA for 3766f88

• [Mosaic TPU] Canonicalize arith.select's condition to vector if other types are vector.

  Show description for 7e2f487
  

  bythew3i

  authored and

  Google-ML-Automation

  committed
  7e2f487

  Copy full SHA for 7e2f487

• Merge pull request jax-ml#24024 from rajasekharporeddy:testbranch1

  Show description for bdae9ac
  Google-ML-Automation

  committed
  bdae9ac

  Copy full SHA for bdae9ac

• Better doc for jnp.cbrt

  rajasekharporeddy

  committed
  5904fe1

  Copy full SHA for 5904fe1

• Add a test for 64-bit precision of IFFT on GPU.

  Show description for ff1c2ac
  

  dfm

  authored and

  Google-ML-Automation

  committed
  ff1c2ac

  Copy full SHA for ff1c2ac

• Update XLA dependency to use revision

  Show description for 504bc43
  Google-ML-Automation

  committed
  504bc43

  Copy full SHA for 504bc43

• Simplify CI configuration.

  Show description for 45cd77a
  

  hawkinsp

  authored and

  Google-ML-Automation

  committed
  45cd77a

  Copy full SHA for 45cd77a

• Move test_aot_device_implicit_transfer to pjit_test.py

  Show description for 203cda6
  

  yashk2810

  authored and

  Google-ML-Automation

  committed
  203cda6

  Copy full SHA for 203cda6

• [pallas:mosaic_gpu] Do not do mgpu.commit_shared if all outputs are invariant wrt sequential axes

  Show description for a046e21
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  a046e21

  Copy full SHA for a046e21

• Remove forward compatibility checks from lowering of LU decomposition.

  Show description for 1a1e16a
  

  dfm

  authored and

  Google-ML-Automation

  committed
  1a1e16a

  Copy full SHA for 1a1e16a

• Rollback because of breakages

  Show description for 411928b
  Google-ML-Automation

  committed
  411928b

  Copy full SHA for 411928b

• Fix and reenable cudnn_fusion_test.

  Show description for b320dc2
  sergachev

  committed
  b320dc2

  Copy full SHA for b320dc2

• [pallas:mosaic_gpu] Do not DCE the jaxpr in the lowering pass

  Show description for b3fca90
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  b3fca90

  Copy full SHA for b3fca90

• [Pallas/MGPU] Undo transforms on refs before giving them back to the users

  Show description for 21fea5b
  

  apaszke

  authored and

  Google-ML-Automation

  committed
  21fea5b

  Copy full SHA for 21fea5b

• Exposed sequential iteration index via pl.program_id in Pallas Mosaic GPU

  Show description for 38d2a57
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  38d2a57

  Copy full SHA for 38d2a57

Commits on Sep 29, 2024

• Update XLA dependency to use revision

  Show description for 2cfbdb6
  Google-ML-Automation

  committed
  2cfbdb6

  Copy full SHA for 2cfbdb6

• Update docs for jnp.trapezoid

  jakevdp

  committed
  7cc5ea4

  Copy full SHA for 7cc5ea4

Commits on Sep 28, 2024

• Update XLA dependency to use revision

  Show description for 6790b90
  Google-ML-Automation

  committed
  6790b90

  Copy full SHA for 6790b90

• Merge pull request jax-ml#23982 from dfm:ffi-call-effects

  Show description for 15024ba
  Google-ML-Automation

  committed
  15024ba

  Copy full SHA for 15024ba

Commits on Sep 27, 2024

• Add support for FFI calls with side effects via ffi_call

  dfm

  committed
  d80a89d

  Copy full SHA for d80a89d

• Bump test tolerance on FFT test that started failing in CI after an XLA change.

  Show description for 061f435
  

  hawkinsp

  authored and

  Google-ML-Automation

  committed
  061f435

  Copy full SHA for 061f435

• Fix test failure when shardy is not enabled.

  Show description for 366c823
  

  hawkinsp

  authored and

  Google-ML-Automation

  committed
  366c823

  Copy full SHA for 366c823

• Fix tests that ask for an accelerator but don't use it.

  Show description for 5969e79
  

  hawkinsp

  authored and

  Google-ML-Automation

  committed
  5969e79

  Copy full SHA for 5969e79

• Merge pull request jax-ml#23957 from jakevdp:choice-doc

  Show description for ff0a98a
  Google-ML-Automation

  committed
  ff0a98a

  Copy full SHA for ff0a98a

• Merge pull request jax-ml#23981 from jakevdp:checkout-pin

  Show description for 4e8a763
  Google-ML-Automation

  committed
  4e8a763

  Copy full SHA for 4e8a763

• Merge pull request jax-ml#23977 from rajasekharporeddy:testbranch1

  Show description for 2963cbe
  Google-ML-Automation

  committed
  2963cbe

  Copy full SHA for 2963cbe

• actions/checkout: pin specific hash

  jakevdp

  committed
  f0e27fd

  Copy full SHA for f0e27fd

• Better documentation for jnp.choose

  jakevdp

  committed
  c061257

  Copy full SHA for c061257

Pagination

Previous

Next