Commits

Branch selector

 jaxlib-v0.4.4

User selector

All users

Datepicker

All time

Commit History

Commits on Feb 16, 2023

• Prepare for jax and jaxlib 0.4.4 release

  Show description for 58e46b4
  

  yashk2810

  authored and

  jax authors

  committed
  58e46b4

  Copy full SHA for 58e46b4

• Remove jax.interpreters.xla.lower_fun.

  Show description for c6a99b6
  

  hawkinsp

  authored and

  jax authors

  committed
  c6a99b6

  Copy full SHA for c6a99b6

• [jax2tf] Enable all native lowering jax2tf tests

  Show description for a9e886f
  

  gnecula

  authored and

  jax authors

  committed
  a9e886f

  Copy full SHA for a9e886f

• [shape_poly] Fix the lowering for symbolic dimension expressions for division

  Show description for 454e4de
  

  gnecula

  authored and

  jax authors

  committed
  454e4de

  Copy full SHA for 454e4de

• Fix the simple bug on call_tf.replace_non_float and add unittest for floating and complex data type.

  Show description for d0b42f2
  

  maxwillzq

  authored and

  jax authors

  committed
  d0b42f2

  Copy full SHA for d0b42f2

• remove `core.{aval_method,aval_property}`

  Show description for 26045c4
  

  froystig

  authored and

  jax authors

  committed
  26045c4

  Copy full SHA for 26045c4

• Merge pull request #14500 from jakevdp:bcsr-matmul-test

  Show description for d8514d0
  

  jax authors

  committed
  d8514d0

  Copy full SHA for d8514d0

• Replace uses of deprecated JAX sharding APIs with their new names in jax.sharding.

  Show description for 0af9fff
  

  hawkinsp

  authored and

  jax authors

  committed
  0af9fff

  Copy full SHA for 0af9fff

• remove `core.axis_substitution_rules`

  Show description for 1b2a318
  

  froystig

  authored and

  jax authors

  committed
  1b2a318

  Copy full SHA for 1b2a318

• Fix pytype errors.

  Show description for 768960b
  

  hawkinsp

  authored and

  jax authors

  committed
  768960b

  Copy full SHA for 768960b

• Remove uses of jax.xla_computation from metadata_test.py

  Show description for 37d4ad9
  

  hawkinsp

  authored and

  jax authors

  committed
  37d4ad9

  Copy full SHA for 37d4ad9

• Merge pull request #14504 from skye:host_callback_pjrt_error

  Show description for 3838d76
  

  jax authors

  committed
  3838d76

  Copy full SHA for 3838d76

• Merge pull request #14479 from skye:cost_analysis

  Show description for fdc6d94
  

  jax authors

  committed
  fdc6d94

  Copy full SHA for fdc6d94

• Merge pull request #14489 from jakevdp:copy-array

  Show description for 6f1527f
  

  jax authors

  committed
  6f1527f

  Copy full SHA for 6f1527f

• MeshComputation.cost_analysis() isn't implemented with PJRT C API.

  Show description for c2819cf
  skye

  committed
  c2819cf

  Copy full SHA for c2819cf

• Raise a user-friendly error message if in/outfeed-based host_callback stuff is used with PJRT C API.

  Show description for d9f628c
  skye

  committed
  d9f628c

  Copy full SHA for d9f628c

Commits on Feb 15, 2023

• [sparse] add bcsr_matmul batching tests

  jakevdp

  committed
  29f91c5

  Copy full SHA for 29f91c5

• Add new argument to jax_test rule

  Show description for 936e4ae
  

  Jake VanderPlas

  authored and

  jax authors

  committed
  936e4ae

  Copy full SHA for 936e4ae

• Merge pull request #14496 from jakevdp:bcsr-concatenate

  Show description for 7fa2470
  

  jax authors

  committed
  7fa2470

  Copy full SHA for 7fa2470

• sparse_test: reduce num_generated_cases to avoid timeouts

  Show description for 6608242
  

  Jake VanderPlas

  authored and

  jax authors

  committed
  6608242

  Copy full SHA for 6608242

• Replace uses of jnp.ndarray with jax.Array inside JAX.

  Show description for cd0533c
  

  hawkinsp

  authored and

  jax authors

  committed
  cd0533c

  Copy full SHA for cd0533c

• Modify JaxArrayTest.test_defragment to work on any numbers of devices

  Show description for 7aa7e15
  

  skye

  authored and

  jax authors

  committed
  7aa7e15

  Copy full SHA for 7aa7e15

• remove `core.bint`

  Show description for 537372a
  

  froystig

  authored and

  jax authors

  committed
  537372a

  Copy full SHA for 537372a

• [sparse] implement bcsr_concatenate

  jakevdp

  committed
  f3e5024

  Copy full SHA for f3e5024

• Merge pull request #14420 from jakevdp:bcoo-broadcast-in-dim

  Show description for 9b288e9
  

  jax authors

  committed
  9b288e9

  Copy full SHA for 9b288e9

• remove `core.{bot,Bot}`

  Show description for 22168a0
  

  froystig

  authored and

  jax authors

  committed
  22168a0

  Copy full SHA for 22168a0

• Merge pull request #14488 from jakevdp:doc-array-methods

  Show description for 7fb4e3b
  

  jax authors

  committed
  7fb4e3b

  Copy full SHA for 7fb4e3b

• [sparse] implement bcsr_broadcast_in_dim

  jakevdp

  committed
  d688b6d

  Copy full SHA for d688b6d

• Merge pull request #14491 from gnecula:call_tf_test

  Show description for 22c1557
  

  jax authors

  committed
  22c1557

  Copy full SHA for 22c1557

• [jax2tf] Add a test to verify that native lowering can be nested inside non-native lowering

  gnecula

  committed
  9adc6e5

  Copy full SHA for 9adc6e5

• DOC: add better documentation for array methods

  jakevdp

  committed
  a6d6858

  Copy full SHA for a6d6858

• [JAX] Deprecate jax.experimental.maps.Mesh.

  Show description for b389eed
  

  hawkinsp

  authored and

  jax authors

  committed
  b389eed

  Copy full SHA for b389eed

• jnp.copy: ensure inputs are array-like

  jakevdp

  committed
  58b800d

  Copy full SHA for 58b800d

• Deprecate uses of jax.experimental.pjit.NamedSharding and jax.experimental.pjit.PartitionSpec.

  Show description for 00d45fe
  

  hawkinsp

  authored and

  jax authors

  committed
  00d45fe

  Copy full SHA for 00d45fe

• Disable some slow tests under asan.

  Show description for 69b8a03
  

  hawkinsp

  authored and

  jax authors

  committed
  69b8a03

  Copy full SHA for 69b8a03

Pagination

Previous

Next