Commits

Branch selector

 12c3057

User selector

All users

Datepicker

All time

Commit History

Commits on Dec 10, 2024

• Introduce lax.ragged_all_to_all primitive

  Show description for 12c3057
  

  ghpvnist

  authored and

  Google-ML-Automation

  committed
  12c3057

  Copy full SHA for 12c3057

• Add a no-op batching rule for optimization_barrier_p

  Show description for 944d822
  

  yashk2810

  authored and

  Google-ML-Automation

  committed
  944d822

  Copy full SHA for 944d822

• Merge pull request jax-ml#25371 from jakevdp:mypy-numpy-version

  Show description for 1743f2c
  Google-ML-Automation

  committed
  1743f2c

  Copy full SHA for 1743f2c

• CI: temporarily pin numpy version for mypy check

  jakevdp

  committed
  a36af96

  Copy full SHA for a36af96

• Merge pull request jax-ml#25368 from hawkinsp:postrelease

  Show description for 7b5cb56
  Google-ML-Automation

  committed
  7b5cb56

  Copy full SHA for 7b5cb56

• Merge branch 'release/0.4.37' into main.

  hawkinsp

  committed
  820f51d

  Copy full SHA for 820f51d

Commits on Dec 9, 2024

• Fix expected exception type in pallas grad tests.

  Show description for 978d35f
  

  dfm

  authored and

  Google-ML-Automation

  committed
  978d35f

  Copy full SHA for 978d35f

• Update XLA dependency to use revision

  Show description for 71c48cb
  Google-ML-Automation

  committed
  71c48cb

  Copy full SHA for 71c48cb

• Port symmetric tridiagonal reduction GPU kernel to FFI.

  Show description for 32df37e
  

  dfm

  authored and

  Google-ML-Automation

  committed
  32df37e

  Copy full SHA for 32df37e

• Update versions for v0.4.37 release.

  hawkinsp

  committed
  ffb07cd

  Copy full SHA for ffb07cd

• Use private names for args in api_util to avoid shadowing kwargs keys.

  Show description for 95892fd
  

  dougalm

  authored and

  hawkinsp

  committed
  95892fd

  Copy full SHA for 95892fd

• Avoid index out of range error in carry structure check

  

  IvyZX

  authored and

  hawkinsp

  committed
  65b6088

  Copy full SHA for 65b6088

• Disable pjit ArrayPjitTest.test_device_put_grad test on TPU v5e

  Show description for 66b9005
  

  kanglant

  authored and

  Google-ML-Automation

  committed
  66b9005

  Copy full SHA for 66b9005

• Merge pull request jax-ml#25272 from justinjfu:pallas_tpu_docs_update

  Show description for 1c07ec6
  Google-ML-Automation

  committed
  1c07ec6

  Copy full SHA for 1c07ec6

• Merge pull request jax-ml#25355 from IvyZX:loop-fix

  Show description for 4db533b
  Google-ML-Automation

  committed
  4db533b

  Copy full SHA for 4db533b

• Merge pull request jax-ml#25294 from jakevdp:array-api-tests

  Show description for dba3358
  Google-ML-Automation

  committed
  dba3358

  Copy full SHA for dba3358

• Merge pull request jax-ml#25351 from jax-ml:dependabot/github_actions/actions/cache-4.2.0

  Show description for 56fcd38
  Google-ML-Automation

  committed
  56fcd38

  Copy full SHA for 56fcd38

• Add error message when using custom_vmap with reverse-mode AD, and add docstrings.

  Show description for 092d2a0
  

  dfm

  authored and

  Google-ML-Automation

  committed
  092d2a0

  Copy full SHA for 092d2a0

• Avoid index out of range error in carry structure check

  IvyZX

  committed
  bd77a70

  Copy full SHA for bd77a70

• [JAX] Add concurrent execution support in colocated Python

  Show description for 296d167
  

  hyeontaek

  authored and

  Google-ML-Automation

  committed
  296d167

  Copy full SHA for 296d167

• Merge pull request jax-ml#25349 from jax-ml:fix-25329

  Show description for d908e0a
  Google-ML-Automation

  committed
  d908e0a

  Copy full SHA for d908e0a

• Bump actions/cache from 4.1.2 to 4.2.0

  Show description for b6863df
  dependabot[bot]

  authored
  b6863df

  Copy full SHA for b6863df

• #sdy enable test_compute_offload_mesh_with_linear_layout for Shardy.

  Show description for 6f69774
  

  bartchr808

  authored and

  Google-ML-Automation

  committed
  6f69774

  Copy full SHA for 6f69774

• Reenable for_loop_test on TPU v5p.

  Show description for f17b2bc
  

  berkinilbeyi

  authored and

  Google-ML-Automation

  committed
  f17b2bc

  Copy full SHA for f17b2bc

• [Mosaic:TPU][NFC] In ext and trunc rules, avoid vreg array reshape by always using implicit shapes

  Show description for b76d264
  

  tlongeri

  authored and

  Google-ML-Automation

  committed
  b76d264

  Copy full SHA for b76d264

• Use private names for args in api_util to avoid shadowing kwargs keys.

  Show description for dd74394
  dougalm

  committed
  dd74394

  Copy full SHA for dd74394

• [Pallas TPU] Improve lowerings for boolean comparison operations

  Show description for 9c98c0c
  

  ayaka14732

  authored and

  Google-ML-Automation

  committed
  9c98c0c

  Copy full SHA for 9c98c0c

• Remove dead code after minimum jaxlib version bump to v0.4.36.

  Show description for 79318a0
  

  hawkinsp

  authored and

  Google-ML-Automation

  committed
  79318a0

  Copy full SHA for 79318a0

• Merge pull request jax-ml#25320 from ROCm:jax-mlgh-9948-fix-kernel-build-upstream

  Show description for cc258f5
  Google-ML-Automation

  committed
  cc258f5

  Copy full SHA for cc258f5

• Ensured that JAX type checks under pytype on Python 3.12

  Show description for 1ac6b76
  

  superbobry

  authored and

  Google-ML-Automation

  committed
  1ac6b76

  Copy full SHA for 1ac6b76

• Merge pull request jax-ml#25338 from carlosgmartin:fix_numpy_linalg_matrix_norm_ord_type_annotation

  Show description for 5a1c4c5
  Google-ML-Automation

  committed
  5a1c4c5

  Copy full SHA for 5a1c4c5

• Activate Tridiagonal Reduction to XLA's FFI

  Show description for d474fed
  

  Paweł Paruzel

  authored and

  Google-ML-Automation

  committed
  d474fed

  Copy full SHA for d474fed

• [Mosaic TPU] Allow downgrading the IR during serialization for forward compat

  Show description for adb2bf6
  

  apaszke

  authored and

  Google-ML-Automation

  committed
  adb2bf6

  Copy full SHA for adb2bf6

• [pallas] Add DotAlgorithmPreset note to CHANGELOG.

  Show description for a94474d
  

  chr1sj0nes

  authored and

  Google-ML-Automation

  committed
  a94474d

  Copy full SHA for a94474d

• [pallas:triton] Add support for DotAlgorithmPreset precision arguments to dot.

  Show description for 3ec55c7
  

  chr1sj0nes

  authored and

  Google-ML-Automation

  committed
  3ec55c7

  Copy full SHA for 3ec55c7

Pagination

Previous

Next