Commits

Branch selector

 487b640

User selector

All users

Datepicker

All time

Commit History

Commits on Jun 22, 2023

• Jax 0.4.13 release.

  

  hawkinsp

  authored and

  skye

  committed
  487b640

  Copy full SHA for 487b640

• Update JAX's XlaExecutable.cost_analysis and related plumbing so it works on Cloud TPU

  Show description for 10424c5
  

  skye

  authored and

  jax authors

  committed
  10424c5

  Copy full SHA for 10424c5

• Silence pytype errors under an upcoming pytype change.

  Show description for 9f4080a
  

  hawkinsp

  authored and

  jax authors

  committed
  9f4080a

  Copy full SHA for 9f4080a

• Merge pull request jax-ml#16508 from hawkinsp:metal

  Show description for e123d1e
  

  jax authors

  committed
  e123d1e

  Copy full SHA for e123d1e

• Ignore JAX_USE_PJRT_C_API_ON_TPU=false user warning raised.

  Show description for 677b0d9
  

  yashk2810

  authored and

  jax authors

  committed
  677b0d9

  Copy full SHA for 677b0d9

• Add a link to the Apple Metal plugin to the JAX README.

  Show description for 85a84fd
  hawkinsp

  committed
  85a84fd

  Copy full SHA for 85a84fd

• Zlib compress kernel proto.

  Show description for b3527f3
  

  chr1sj0nes

  authored and

  jax authors

  committed
  b3527f3

  Copy full SHA for b3527f3

• Make JAX-Triton calls serializable.

  Show description for f238667
  

  chr1sj0nes

  authored and

  jax authors

  committed
  f238667

  Copy full SHA for f238667

• [jax2tf] Refactor the backwards compatibility tests.

  Show description for c5a47d1
  

  gnecula

  authored and

  jax authors

  committed
  c5a47d1

  Copy full SHA for c5a47d1

• [shape_poly] linalg.cholesky: add support for shape polymorphism with native lowering on CPU

  Show description for f02d122
  

  gnecula

  authored and

  jax authors

  committed
  f02d122

  Copy full SHA for f02d122

• [shape_poly] linalg.qr: shape polymorphism with native lowering on CPU

  Show description for d940db6
  

  gnecula

  authored and

  jax authors

  committed
  d940db6

  Copy full SHA for d940db6

• [shape_poly] linalg.svd: shape polymorphism for native serialization on CPU

  Show description for 92288d3
  

  gnecula

  authored and

  jax authors

  committed
  92288d3

  Copy full SHA for 92288d3

• Documentation improvements for XlaCallModule.disabled_checks.

  Show description for 1535fa0
  

  gnecula

  authored and

  jax authors

  committed
  1535fa0

  Copy full SHA for 1535fa0

• Merge pull request jax-ml#16499 from skye:se_warning

  Show description for 5742ab4
  

  jax authors

  committed
  5742ab4

  Copy full SHA for 5742ab4

Commits on Jun 21, 2023

• Use EncapsulateFunction utility.

  Show description for 64e7327
  

  chr1sj0nes

  authored and

  jax authors

  committed
  64e7327

  Copy full SHA for 64e7327

• Fix jax.clear_backends() on Cloud TPU

  Show description for 529822c
  

  skye

  authored and

  jax authors

  committed
  529822c

  Copy full SHA for 529822c

• Merge pull request jax-ml#16484 from jakevdp:key-shape

  Show description for f46e514
  

  jax authors

  committed
  f46e514

  Copy full SHA for f46e514

• Merge pull request jax-ml#16404 from jakevdp:tracer-errors

  Show description for 06f76bc
  

  jax authors

  committed
  06f76bc

  Copy full SHA for 06f76bc

Commits on Jun 20, 2023

• Merge pull request jax-ml#16333 from skye:profiler_test

  Show description for 961d918
  

  jax authors

  committed
  961d918

  Copy full SHA for 961d918

• profiler_test.py fixes and add coverage to Cloud TPU CI

  Show description for 2ca151e
  skye

  committed
  2ca151e

  Copy full SHA for 2ca151e

• Issue warning if JAX_USE_PJRT_C_API_ON_TPU=false

  Show description for 71cbfe4
  skye

  committed
  71cbfe4

  Copy full SHA for 71cbfe4

• Merge pull request jax-ml#16498 from ROCmSoftwarePlatform:rocm-update-build-defaults

  Show description for 3cfb57d
  

  jax authors

  committed
  3cfb57d

  Copy full SHA for 3cfb57d

• Fix symbol collision when merging MLIR modules.

  Show description for e99ca46
  

  hawkinsp

  authored and

  jax authors

  committed
  e99ca46

  Copy full SHA for e99ca46

• Merge pull request jax-ml#16497 from hawkinsp:workflows

  Show description for eca3b97
  

  jax authors

  committed
  eca3b97

  Copy full SHA for eca3b97

• Rename Github actions so they have more consistent naming.

  hawkinsp

  committed
  741f0d0

  Copy full SHA for 741f0d0

• [ROCm]: Updates defaults in build script

  

  Rahul Batra

  committed
  88a2c8c

  Copy full SHA for 88a2c8c

• Fix test failure in array_interoperability_test due to 64-bit dtype squashing.

  Show description for b842e86
  

  hawkinsp

  authored and

  jax authors

  committed
  b842e86

  Copy full SHA for b842e86

• Merge pull request jax-ml#16489 from hawkinsp:plugins2

  Show description for 8f83da6
  

  jax authors

  committed
  8f83da6

  Copy full SHA for 8f83da6

• Add a warning if a non-allowlisted plugin is used.

  Show description for 5d665a5
  hawkinsp

  committed
  5d665a5

  Copy full SHA for 5d665a5

• [XLA:Python] Fix __cuda_array_interface__.

  Show description for 0ec03db
  

  hawkinsp

  authored and

  jax authors

  committed
  0ec03db

  Copy full SHA for 0ec03db

• Merge pull request jax-ml#16457 from hawkinsp:docs

  Show description for afcd1a7
  

  jax authors

  committed
  afcd1a7

  Copy full SHA for afcd1a7

• Merge pull request jax-ml#16488 from hawkinsp:winfx

  Show description for 018ea34
  

  jax authors

  committed
  018ea34

  Copy full SHA for 018ea34

• Add a CI presubmit that renders the documentation.

  hawkinsp

  committed
  34d9f5a

  Copy full SHA for 34d9f5a

• Fix xla_bridge_test test failures under Windows.

  hawkinsp

  committed
  cc0bbde

  Copy full SHA for cc0bbde

• Merge pull request jax-ml#16437 from ROCmSoftwarePlatform:rocm-updates-build

  Show description for 632c790
  

  jax authors

  committed
  632c790

  Copy full SHA for 632c790

Pagination

Previous

Next