Commits

Branch selector

 f16ea4f

User selector

All users

Datepicker

All time

Commit History

Commits on Oct 20, 2023

• Fix oldest supported libtpu install command

  skye

  committed
  f16ea4f

  Copy full SHA for f16ea4f

Commits on Oct 17, 2023

• Merge pull request jax-ml#18070 from sagyakwa:patch-1

  Show description for 8353027
  

  jax authors

  committed
  8353027

  Copy full SHA for 8353027

• Add GetOpSharding to XLA/PjRt utils.

  Show description for 7498377
  

  jax authors

  committed
  7498377

  Copy full SHA for 7498377

• Merge branch 'google:main' into patch-1

  sagyakwa

  authored
  e9da4f5

  Copy full SHA for e9da4f5

• random_lax_test: Bump shards number for CPU config.

  Show description for d03bbc0
  

  jax authors

  committed
  d03bbc0

  Copy full SHA for d03bbc0

• [Pallas TPU] Add DMA descriptor abstraction for constructing but not starting DMAs

  Show description for 86023f5
  

  sharadmv

  authored and

  jax authors

  committed
  86023f5

  Copy full SHA for 86023f5

• [pallas:gpu] Simplify broadcast_to, min, max lowering.

  Show description for c16b893
  

  chr1sj0nes

  authored and

  jax authors

  committed
  c16b893

  Copy full SHA for c16b893

• [Mosaic] apply_vector_layout C++ rewrite: add tpu.concatenate

  Show description for 2c9ea51
  

  tlongeri

  authored and

  jax authors

  committed
  2c9ea51

  Copy full SHA for 2c9ea51

• Make sure we don't filter stack frames of packages that start with a jax prefix

  Show description for b84ae98
  

  apaszke

  authored and

  jax authors

  committed
  b84ae98

  Copy full SHA for b84ae98

• Rollback to fix internal breakage

  Show description for 2be6019
  

  jax authors

  committed
  2be6019

  Copy full SHA for 2be6019

• Update XLA dependency to use revision

  Show description for 2604d0c
  

  jax authors

  committed
  2604d0c

  Copy full SHA for 2604d0c

• Re-enable the test as the GPU plugin profiler fixes are in.

  Show description for 497f809
  

  Jieying Luo

  authored and

  jax authors

  committed
  497f809

  Copy full SHA for 497f809

• Merge pull request jax-ml#18117 from jakevdp:ad-deprecations

  Show description for 6540810
  

  jax authors

  committed
  6540810

  Copy full SHA for 6540810

• Merge pull request jax-ml#18105 from jakevdp:keyarray

  Show description for 7d203ae
  

  jax authors

  committed
  7d203ae

  Copy full SHA for 7d203ae

Commits on Oct 16, 2023

• Temporarily set TPU_LIBRARY_PATH in xla_bridge.

  Show description for 43fc423
  

  Jieying Luo

  authored and

  jax authors

  committed
  43fc423

  Copy full SHA for 43fc423

• [XLA:GPU] Fix bug in all-to-all for complex data types.

  Show description for 89b5449
  

  hawkinsp

  authored and

  jax authors

  committed
  89b5449

  Copy full SHA for 89b5449

• Merge pull request jax-ml#18104 from gnecula:multi_jax2tf

  Show description for 5919c1f
  

  jax authors

  committed
  5919c1f

  Copy full SHA for 5919c1f

• Merge pull request jax-ml#18120 from fehiepsi:slogdet

  Show description for f285359
  

  jax authors

  committed
  f285359

  Copy full SHA for f285359

• Fix testProfilerGetFDOProfile.

  Show description for 93fbf62
  

  wang12tao

  authored and

  jax authors

  committed
  93fbf62

  Copy full SHA for 93fbf62

• [pallas] dot fixes.

  Show description for dcc92e3
  

  chr1sj0nes

  authored and

  jax authors

  committed
  dcc92e3

  Copy full SHA for dcc92e3

• jax.interpreters.ad: deprecate inadvertently exported symbols

  jakevdp

  committed
  0eef3b5

  Copy full SHA for 0eef3b5

• [pallas:gpu] Directly use Triton built-in function lowering for max_contiguous and multiple_of.

  Show description for 675cb15
  

  chr1sj0nes

  authored and

  jax authors

  committed
  675cb15

  Copy full SHA for 675cb15

• [pallas:gpu] Simplify reshape lowering rule.

  Show description for ac2c228
  

  chr1sj0nes

  authored and

  jax authors

  committed
  ac2c228

  Copy full SHA for ac2c228

• Build jaxlib without PJRT GPU deps when plugin will be built.

  Show description for 0290150
  

  Jieying Luo

  authored and

  jax authors

  committed
  0290150

  Copy full SHA for 0290150

• [jax2tf] First step to enable multi-platform native lowering

  Show description for b65c1b2
  gnecula

  committed
  b65c1b2

  Copy full SHA for b65c1b2

• Update XLA dependency to use revision

  Show description for 110b8d7
  

  jax authors

  committed
  110b8d7

  Copy full SHA for 110b8d7

Commits on Oct 15, 2023

• Update XLA dependency to use revision

  Show description for 8bf605a
  

  jax authors

  committed
  8bf605a

  Copy full SHA for 8bf605a

Commits on Oct 14, 2023

• Merge pull request jax-ml#18112 from jakevdp:broadcast-shape

  Show description for f9e4ab7
  

  jax authors

  committed
  f9e4ab7

  Copy full SHA for f9e4ab7

• Merge pull request jax-ml#18111 from superbobry:fix-flags-underscore

  Show description for 8d700df
  

  jax authors

  committed
  8d700df

  Copy full SHA for 8d700df

• Enhance the speed of slogdet with qr method

  fehiepsi

  committed
  c97ecf4

  Copy full SHA for c97ecf4

• Update XLA dependency to use revision

  Show description for 9a3ec22
  

  jax authors

  committed
  9a3ec22

  Copy full SHA for 9a3ec22

Commits on Oct 13, 2023

• Merge pull request jax-ml#18110 from jakevdp:jex-redirect

  Show description for 12c2baa
  

  jax authors

  committed
  12c2baa

  Copy full SHA for 12c2baa

• [typing] allow scalar shape for jnp.broadcast_to

  jakevdp

  committed
  1815bc7

  Copy full SHA for 1815bc7

• MAINT Drop underscore from the name of externally-referenced state objects

  superbobry

  committed
  f9087ab

  Copy full SHA for f9087ab

• Disable pgle_test.py for GPU plugin.

  Show description for 4cd4f3f
  

  Jieying Luo

  authored and

  jax authors

  committed
  4cd4f3f

  Copy full SHA for 4cd4f3f

Pagination

Previous

Next