Commits

Branch selector

 main

User selector

skye

Datepicker

All time

Commit History

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

Commits on Aug 6, 2021

• Don't try to use persistent compilation cache when running CPU computations.

  skye

  committed
  8190286

  Copy full SHA for 8190286

Commits on Aug 5, 2021

• Update WORKSPACE

  Show description for 40ccad8
  

  skye

  authored and

  jax authors

  committed
  40ccad8

  Copy full SHA for 40ccad8

• Scrub function pointers in op_name metadata in persistent compilation cache key

  Show description for dcf3712
  skye

  committed
  dcf3712

  Copy full SHA for dcf3712

Commits on Jul 30, 2021

• Add jax_hlo_source_file_canonicalization_regex config.

  Show description for 350045c
  skye

  committed
  350045c

  Copy full SHA for 350045c

Commits on Jul 21, 2021

• Bump jax version and CHANGELOG to 0.2.18

  skye

  committed
  a7916f1

  Copy full SHA for a7916f1

Commits on Jul 13, 2021

• Changes to make jax[tpu] work better in a docker container.

  Show description for b2fd6a7
  skye

  committed
  b2fd6a7

  Copy full SHA for b2fd6a7

Commits on Jul 12, 2021

• Update generate_release_index[es].py to also produce libtpu_releases.html.

  Show description for 1a650d2
  skye

  committed
  1a650d2

  Copy full SHA for 1a650d2

Commits on Jul 10, 2021

• Add config.define_string_state.

  Show description for 3006756
  skye

  committed
  3006756

  Copy full SHA for 3006756

Commits on Jun 23, 2021

• Update jax version to 0.2.16

  skye

  committed
  3da8a4c

  Copy full SHA for 3da8a4c

• Fix pip install jax[tpu]

  Show description for 55276d1
  skye

  committed
  55276d1

  Copy full SHA for 55276d1

• Revert "Improve support for pip install jax[cuda111]"

  Show description for 2460f91
  skye

  committed
  2460f91

  Copy full SHA for 2460f91

• Update jax version to 0.2.15

  skye

  committed
  444ee5e

  Copy full SHA for 444ee5e

• Add pip install jax[tpu] configuration.

  Show description for a12a229
  skye

  committed
  a12a229

  Copy full SHA for a12a229

Commits on Jun 22, 2021

• On Cloud TPU, use pip-installed libtpu instead of system default if applicable.

  skye

  committed
  ba972f0

  Copy full SHA for ba972f0

Commits on Jun 15, 2021

• Add jaxlib version check on newly-added NumpyLinalgTest.testEighZeroDiagonal test.

  Show description for 63355a9
  skye

  committed
  63355a9

  Copy full SHA for 63355a9

Commits on Jun 10, 2021

• Update jax version to 0.2.14

  skye

  committed
  063401f

  Copy full SHA for 063401f

• Pin the tpu_driver version used for Cloud TPU Colabs, instead of using nightly.

  Show description for 4abac4f
  skye

  committed
  4abac4f

  Copy full SHA for 4abac4f

Commits on Jun 3, 2021

• Clean up multi-process guide language.

  Show description for 4a85877
  skye

  committed
  4a85877

  Copy full SHA for 4a85877

Commits on Jun 2, 2021

• Update cloud_tpu_colabs README to reflect Cloud TPU VM announcement.

  skye

  committed
  8a65ff8

  Copy full SHA for 8a65ff8

Commits on May 21, 2021

• Add multi-process guide to docs.

  Show description for 40cc1ec
  skye

  committed
  40cc1ec

  Copy full SHA for 40cc1ec

Pagination

Previous

Next