Commits

Branch selector

 0ce7c7b

User selector

All users

Datepicker

All time

Commit History

Commits on Nov 8, 2023

• Register plugin profiler for TPU and remove --config=tpu/--enable_tpu in jaxlib.

  Show description for 0ce7c7b
  

  Jieying Luo

  authored and

  jax authors

  committed
  0ce7c7b

  Copy full SHA for 0ce7c7b

• Update XLA dependency to use revision

  Show description for 7f08a0d
  

  jax authors

  committed
  7f08a0d

  Copy full SHA for 7f08a0d

• [Mosaic] Expose C API for apply-vector-layout's assemble, disassemble, relayout and applyLayoutOp

  Show description for 250486f
  

  tlongeri

  authored and

  jax authors

  committed
  250486f

  Copy full SHA for 250486f

• [Mosaic][NFC] C++ apply_vector_layout: Do not pass RewriteContext to functions that only need target_shape

  Show description for 52b0141
  

  tlongeri

  authored and

  jax authors

  committed
  52b0141

  Copy full SHA for 52b0141

• Small Pallas bug fixes.

  Show description for 2b6be33
  

  epiqueras

  authored and

  jax authors

  committed
  2b6be33

  Copy full SHA for 2b6be33

Commits on Nov 7, 2023

• Fix test for distributed backend initialization.

  Show description for 647afbb
  

  hawkinsp

  authored and

  jax authors

  committed
  647afbb

  Copy full SHA for 647afbb

• [XLA:Mosaic] Add support for arith.bitcast for both scalar and vector.

  Show description for ecceac3
  

  bythew3i

  authored and

  jax authors

  committed
  ecceac3

  Copy full SHA for ecceac3

• Add a regression test for a pmap issue that is fixed at head.

  Show description for f4eb3f6
  

  hawkinsp

  authored and

  jax authors

  committed
  f4eb3f6

  Copy full SHA for f4eb3f6

• [Mosaic] Add support for extracting the first element of a vector as a scalar

  Show description for e66f4e9
  

  apaszke

  authored and

  jax authors

  committed
  e66f4e9

  Copy full SHA for e66f4e9

• Move test for backend initialization into jax.distributed.initialize() wrapper.

  Show description for b85ea68
  

  hawkinsp

  authored and

  jax authors

  committed
  b85ea68

  Copy full SHA for b85ea68

• [Mosaic] apply_vector_layout C++: Fix check in vector.broadcast rule

  Show description for c5d6df4
  

  tlongeri

  authored and

  jax authors

  committed
  c5d6df4

  Copy full SHA for c5d6df4

• Update XLA dependency to use revision

  Show description for 5a1731c
  

  jax authors

  committed
  5a1731c

  Copy full SHA for 5a1731c

• Copybara import of the project:

  Show description for abf2c5c
  

  chaserileyroberts

  authored and

  jax authors

  committed
  abf2c5c

  Copy full SHA for abf2c5c

• Move code around to make _mha_backward xla implementation avoid unnecessary kernel computation.

  Show description for 5b1edac
  

  zhangqiaorjc

  authored and

  jax authors

  committed
  5b1edac

  Copy full SHA for 5b1edac

Commits on Nov 6, 2023

• Merge pull request jax-ml#18415 from jakevdp:cumprod-doc

  Show description for 5b534ff
  

  jax authors

  committed
  5b534ff

  Copy full SHA for 5b534ff

• Merge pull request jax-ml#18413 from jakevdp:bcoo-error

  Show description for fc8edcf
  

  jax authors

  committed
  fc8edcf

  Copy full SHA for fc8edcf

• Merge pull request jax-ml#18414 from jakevdp:optimize-upstream-fix

  Show description for 8889af0
  

  jax authors

  committed
  8889af0

  Copy full SHA for 8889af0

• Fix dtype-related docs for cumsum/cumprod

  jakevdp

  committed
  e793285

  Copy full SHA for e793285

• Make jax.scipy.optimize test compatible with upstream scipy

  jakevdp

  committed
  17a2623

  Copy full SHA for 17a2623

• Fix the missing cache_misses metric when min compile time is set to zero.

  Show description for 7e37294
  

  jax authors

  committed
  7e37294

  Copy full SHA for 7e37294

• [sparse] fix concretization error for nse

  jakevdp

  committed
  68e6bc0

  Copy full SHA for 68e6bc0

• Merge pull request jax-ml#18411 from 8bitmp3:add_kaggle

  Show description for 5f4d479
  

  jax authors

  committed
  5f4d479

  Copy full SHA for 5f4d479

• Raise an exception if jax.distributed.initialize() is called after backends have been initialized.

  Show description for eeafff5
  

  hawkinsp

  authored and

  jax authors

  committed
  eeafff5

  Copy full SHA for eeafff5

• Merge pull request jax-ml#11625 from jakevdp:deprecate-shuffle

  Show description for f54be7a
  

  jax authors

  committed
  f54be7a

  Copy full SHA for f54be7a

• Switch to the new cache-key generation algorithm.

  Show description for 28e33ca
  

  jax authors

  committed
  28e33ca

  Copy full SHA for 28e33ca

• Merge pull request jax-ml#18072 from jakevdp:spec-0

  Show description for 5e24f27
  

  jax authors

  committed
  5e24f27

  Copy full SHA for 5e24f27

• Merge pull request jax-ml#18378 from jakevdp:install-docs

  Show description for 5ec922a
  

  jax authors

  committed
  5ec922a

  Copy full SHA for 5ec922a

• Add Kaggle integration for JAX pmap cookbook

  8bitmp3

  committed
  ed2fef8

  Copy full SHA for ed2fef8

• Deprecate jax.random.shuffle

  Show description for 5f7335f
  jakevdp

  committed
  5f7335f

  Copy full SHA for 5f7335f

• Merge pull request jax-ml#18380 from jakevdp:cross-2d

  Show description for 29a6262
  

  jax authors

  committed
  29a6262

  Copy full SHA for 29a6262

• Change deprecation policy from NEP 29 to SPEC 0

  jakevdp

  committed
  57d69c8

  Copy full SHA for 57d69c8

• Merge pull request jax-ml#18379 from jakevdp:nanmean-dtype

  Show description for 189d3ab
  

  jax authors

  committed
  189d3ab

  Copy full SHA for 189d3ab

• Add cudnn version to the jax-cuda-pjrt version.

  Show description for 0f1619e
  

  Jieying Luo

  authored and

  jax authors

  committed
  0f1619e

  Copy full SHA for 0f1619e

• [Mosaic] Don't link CAPIIR into _tpu_ext, link into jaxlib_mlir_capi_shared_library instead

  Show description for c186928
  

  tlongeri

  authored and

  jax authors

  committed
  c186928

  Copy full SHA for c186928

• Fix jax2tf fft test failure.

  Show description for 41430ac
  

  hawkinsp

  authored and

  jax authors

  committed
  41430ac

  Copy full SHA for 41430ac

Pagination

Previous

Next