Commits

Branch selector

 main

User selector

skye

Datepicker

All time

Commit History

Commits on Jun 14, 2022

• Add jax.config.jax_default_device to jax in-memory cache key

  Show description for 7098088
  

  skye

  authored and

  jax authors

  committed
  7098088

  Copy full SHA for 7098088

Commits on Jun 8, 2022

• Add jax.default_device to CHANGELOG

  skye

  committed
  f862825

  Copy full SHA for f862825

Commits on May 16, 2022

• Update version + CHANGELOG for jax 0.3.13 release

  skye

  committed
  6b926d5

  Copy full SHA for 6b926d5

• Update xla_client._version and add missing version checks to JAX

  Show description for 744f6b4
  

  skye

  authored and

  jax authors

  committed
  744f6b4

  Copy full SHA for 744f6b4

Commits on May 14, 2022

• Improve pjit logging when JAX_DISTRIBUTED_DEBUG=1 is enabled.

  Show description for 5f26422
  skye

  committed
  5f26422

  Copy full SHA for 5f26422

Commits on May 7, 2022

• Add jax.default_device context manager

  Show description for f26b866
  skye

  committed
  f26b866

  Copy full SHA for f26b866

Commits on Mar 16, 2022

• Bump jax and jaxlib versions for 0.3.2 release

  Show description for d7087ab
  skye

  committed
  d7087ab

  Copy full SHA for d7087ab

• Update CHANGELOG and setup.py for jax + jaxlib 0.3.2 releases

  skye

  committed
  f9775a2

  Copy full SHA for f9775a2

Commits on Mar 15, 2022

• Add TPU v4 support to cloud_tpu_init

  Show description for b36cd25
  skye

  committed
  b36cd25

  Copy full SHA for b36cd25

• Update WORKSPACE for jaxlib 0.3.2 release, take 3

  Show description for 057905b
  

  skye

  authored and

  jax authors

  committed
  057905b

  Copy full SHA for 057905b

Commits on Mar 14, 2022

• Update WORKSPACE for jaxlib 0.3.2, take 2

  skye

  committed
  1ba5178

  Copy full SHA for 1ba5178

Commits on Mar 11, 2022

• Update jaxlib version to 0.3.2 to match jax

  skye

  committed
  5c8c4d4

  Copy full SHA for 5c8c4d4

• Update WORKSPACE for jaxlib 0.3.1

  skye

  committed
  ddd7477

  Copy full SHA for ddd7477

Commits on Mar 8, 2022

• Improve TPU v2 and v3 mesh_utils.create_device_mesh logic.

  Show description for bcee442
  skye

  committed
  bcee442

  Copy full SHA for bcee442

Commits on Mar 7, 2022

• Remove pjit warning.

  Show description for afca0b5
  skye

  committed
  afca0b5

  Copy full SHA for afca0b5

Commits on Feb 4, 2022

• Run compilation_cache_test.py using TFRT

  Show description for 3fd91d6
  

  skye

  authored and

  jax authors

  committed
  3fd91d6

  Copy full SHA for 3fd91d6

Commits on Feb 2, 2022

• Bump jax version to 0.2.29 and update CHANGELOG

  skye

  committed
  715066c

  Copy full SHA for 715066c

Commits on Jan 29, 2022

• Update pinned libtpu-nightly version for jaxlib 0.1.76

  skye

  committed
  d096d9a

  Copy full SHA for d096d9a

Commits on Dec 14, 2021

• Handle the edge case where two pjit.FROM_GDA singletons are created.

  skye

  committed
  6bdeb82

  Copy full SHA for 6bdeb82

Commits on Dec 10, 2021

• Add contiguous_submeshes option to mesh_utils.create_device_mesh().

  Show description for 17b0866
  skye

  committed
  17b0866

  Copy full SHA for 17b0866

Commits on Nov 12, 2021

• Add option to key on extra command-line flags in persistent compilation cache.

  Show description for 649eab3
  

  skye

  authored and

  jax authors

  committed
  649eab3

  Copy full SHA for 649eab3

Commits on Oct 13, 2021

• Update jax version number.

  Show description for 2cd6fa3
  skye

  committed
  2cd6fa3

  Copy full SHA for 2cd6fa3

• Update jax version and CHANGELOG for 0.2.22 release

  skye

  committed
  962c496

  Copy full SHA for 962c496

• Update CHANGELOG and verson numbers for jaxlib 0.1.72 release

  skye

  committed
  0072c32

  Copy full SHA for 0072c32

Commits on Oct 12, 2021

• Update WORKSPACE for jaxlib 0.1.72 release

  skye

  committed
  49a3b37

  Copy full SHA for 49a3b37

Commits on Oct 6, 2021

• Include XLA_FLAGS in persistent compilation cache key.

  Show description for f939048
  skye

  committed
  f939048

  Copy full SHA for f939048

Commits on Oct 5, 2021

• Convert 4-space indents to 2-space.

  skye

  committed
  62c922a

  Copy full SHA for 62c922a

Commits on Sep 30, 2021

• Mark --jax_xla_backend as deprecated.

  skye

  committed
  8488903

  Copy full SHA for 8488903

• Add --jax_platforms flag to replace --jax_platform_name.

  Show description for adcab94
  skye

  committed
  adcab94

  Copy full SHA for adcab94

Commits on Sep 29, 2021

• Add jaxlib version check to pjit_test.py

  skye

  committed
  7a848a3

  Copy full SHA for 7a848a3

Commits on Sep 28, 2021

• Refactor xla_bridge.get_backend().

  Show description for 5400bf4
  skye

  committed
  5400bf4

  Copy full SHA for 5400bf4

• Add xla_bridge_test.GetBackendTest

  Show description for 836926c
  skye

  committed
  836926c

  Copy full SHA for 836926c

Commits on Sep 23, 2021

• Remove .[minimum-jaxlib] from test-requirements.txt

  Show description for 2fcf3f7
  skye

  committed
  2fcf3f7

  Copy full SHA for 2fcf3f7

Commits on Sep 7, 2021

• Unpin the tpu_driver version used for Cloud TPU Colabs.

  Show description for 706ba0e
  skye

  committed
  706ba0e

  Copy full SHA for 706ba0e

Commits on Aug 16, 2021

• Factor out uncached version of xla_bridge.get_backend

  Show description for 4e6aef5
  

  skye

  authored and

  jax authors

  committed
  4e6aef5

  Copy full SHA for 4e6aef5

Pagination

Previous

Next