Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Dec 17, 2023

• Merge pull request #19017 from gnecula:export_shard_map

  Show description for e0cc987
  

  jax authors

  committed
  e0cc987

  Copy full SHA for e0cc987

• [export] Add test for exporting jit(shard_map)

  gnecula

  committed
  d1e5199

  Copy full SHA for d1e5199

• Update XLA dependency to use revision

  Show description for 7af7285
  

  jax authors

  committed
  7af7285

  Copy full SHA for 7af7285

Commits on Dec 16, 2023

• Update XLA dependency to use revision

  Show description for 6d876ed
  

  jax authors

  committed
  6d876ed

  Copy full SHA for 6d876ed

• Merge pull request #19012 from skye:test_no_log_spam

  Show description for 732fd15
  

  jax authors

  committed
  732fd15

  Copy full SHA for 732fd15

• Set `TF_CPP_MIN_LOG_LEVEL=1` for new log spam test.

  skye

  committed
  904103e

  Copy full SHA for 904103e

Commits on Dec 15, 2023

• Bump oldest support libtpu version to 20230927 which includes CopyToMemorySpace.

  Show description for 717059e
  

  Jieying Luo

  authored and

  jax authors

  committed
  717059e

  Copy full SHA for 717059e

• Merge pull request #19005 from jakevdp:array-api-linalg

  Show description for bc239fd
  

  jax authors

  committed
  bc239fd

  Copy full SHA for bc239fd

• Merge pull request #19006 from skye:test_no_log_spam

  Show description for 65d2f36
  

  jax authors

  committed
  65d2f36

  Copy full SHA for 65d2f36

• Add unit test for catching log spam.

  skye

  committed
  e5f7598

  Copy full SHA for e5f7598

• jnp.linalg: add matrix_norm, matrix_transpose, vector_norm, vector_transpose

  Show description for 0c7b959
  jakevdp

  committed
  0c7b959

  Copy full SHA for 0c7b959

• Merge pull request #19002 from jakevdp:core-dep

  Show description for 64799a4
  

  jax authors

  committed
  64799a4

  Copy full SHA for 64799a4

• jax.core: deprecate several private APIs

  jakevdp

  committed
  adefbca

  Copy full SHA for adefbca

• Merge pull request #19000 from jakevdp:array-api-update

  Show description for 9462aec
  

  jax authors

  committed
  9462aec

  Copy full SHA for 9462aec

• array-api: update CI to use latest array-api-tests commit

  jakevdp

  committed
  ad67726

  Copy full SHA for ad67726

• Merge pull request #18989 from hawkinsp:fork

  Show description for eb595af
  

  jax authors

  committed
  eb595af

  Copy full SHA for eb595af

• Add two flags to support only building cuda kernel plugin or cuda pjrt plugin.

  Show description for c8b3567
  

  Jieying Luo

  authored and

  jax authors

  committed
  c8b3567

  Copy full SHA for c8b3567

• Add a warning if the user calls os.fork().

  Show description for ec89e5e
  hawkinsp

  committed
  ec89e5e

  Copy full SHA for ec89e5e

• Rolling back #18980, because it is not backwards compatible and breaks existing users.

  Show description for 4153112
  

  superbobry

  authored and

  jax authors

  committed
  4153112

  Copy full SHA for 4153112

• Import new version of Triton

  Show description for a0458e6
  

  gflegar

  authored and

  jax authors

  committed
  a0458e6

  Copy full SHA for a0458e6

• Merge pull request #18980 from gnecula:export_api

  Show description for 91faddd
  

  jax authors

  committed
  91faddd

  Copy full SHA for 91faddd

• [export] Refactor the imports for the public API of jax.experimental.export

  Show description for fd0f007
  gnecula

  committed
  fd0f007

  Copy full SHA for fd0f007

• Update XLA dependency to use revision

  Show description for 14ce5f7
  

  jax authors

  committed
  14ce5f7

  Copy full SHA for 14ce5f7

• Merge pull request #18956 from gnecula:export_effects

  Show description for 6cd7ada
  

  jax authors

  committed
  6cd7ada

  Copy full SHA for 6cd7ada

• [export] Fix the serialization of effects

  Show description for 552010a
  gnecula

  committed
  552010a

  Copy full SHA for 552010a

• Merge pull request #18991 from skye:revert_cuda_install

  Show description for a7b6023
  

  jax authors

  committed
  a7b6023

  Copy full SHA for a7b6023

• Revert "Recommend the plugin in the CUDA installation instructions."

  Show description for 5d26c30
  skye

  committed
  5d26c30

  Copy full SHA for 5d26c30

• Merge pull request #18983 from mattjj:eager-pmap-custom-jvp-vmap

  Show description for 891d44c
  

  jax authors

  committed
  891d44c

  Copy full SHA for 891d44c

• fix a bug with eager pmap + vmap + custom_jvp interaction

  Show description for b70ac90
  mattjj

  committed
  b70ac90

  Copy full SHA for b70ac90

Commits on Dec 14, 2023

• Merge pull request #18987 from hawkinsp:plugin

  Show description for c4a9cee
  

  jax authors

  committed
  c4a9cee

  Copy full SHA for c4a9cee

• Remove local version in jax-cuda-plugin and jax-cuda-pjrt package.

  Show description for 1559d64
  

  Jieying Luo

  authored and

  jax authors

  committed
  1559d64

  Copy full SHA for 1559d64

• Recommend the plugin in the CUDA installation instructions.

  hawkinsp

  committed
  78bc4f5

  Copy full SHA for 78bc4f5

• Remove XLA logging patch, now the XLA fix has landed.

  Show description for cf5a495
  

  hawkinsp

  authored and

  jax authors

  committed
  cf5a495

  Copy full SHA for cf5a495

• [roll forward 2] Remove the `jax_require_devices_during_lowering flag since it was temporary. Added the semi-breaking change to Changelog.md.

  Show description for 8bf3a86
  

  yashk2810

  authored and

  jax authors

  committed
  8bf3a86

  Copy full SHA for 8bf3a86

• Finish release of jax and jaxlib 0.4.23

  Show description for 6e1ab7c
  

  yashk2810

  authored and

  jax authors

  committed
  6e1ab7c

  Copy full SHA for 6e1ab7c

Pagination

Previous

Next