Commits

Branch selector

 master

User selector

All users

Datepicker

All time

Commit History

Commits on Feb 26, 2023

• [jax2tf] Include more sharding annotations in the TF graph

  Show description for 0cdb7f9
  

  gnecula

  authored and

  jax authors

  committed
  0cdb7f9

Commits on Feb 25, 2023

• Merge pull request jax-ml#14684 from sharadmv:flake-fix

  Show description for 7217686
  

  jax authors

  committed
  7217686

• Remove TokenSet needing to have effects in a certain order

  sharadmv

  committed
  18c6cbe

• Merge pull request jax-ml#14614 from sharadmv:ref

  Show description for 8ebfb0b
  

  jax authors

  committed
  8ebfb0b

• Remove jax._src deletion.

  Show description for b61d5d5
  

  hawkinsp

  authored and

  jax authors

  committed
  b61d5d5

• lax_scipy_test: Revert split into three targets.

  Show description for 0292f5d
  

  pizzud

  authored and

  jax authors

  committed
  0292f5d

Commits on Feb 24, 2023

• Remove use_stablehlo as minimum mlir_api_version >= 43

  Show description for d84ac22
  

  yashk2810

  authored and

  jax authors

  committed
  d84ac22

• Bump minimum jaxlib version to 0.4.4 which means xla_extension_version >= 127

  Show description for aa5e229
  

  yashk2810

  authored and

  jax authors

  committed
  aa5e229

• BUG: raise error when shaped_abstractify is called on JAX scalar types

  Show description for 7f68266
  

  Jake VanderPlas

  authored and

  jax authors

  committed
  7f68266

• Create avals and pass them to _check_sharding rather than the actual value.

  Show description for d277358
  

  yashk2810

  authored and

  jax authors

  committed
  d277358

• Merge pull request jax-ml#13756 from mattjj:remat-docs

  Show description for d12cdc6
  

  jax authors

  committed
  d12cdc6

• Merge pull request jax-ml#14650 from jakevdp:bcoo-matmat-grad-fast

  Show description for 92cdb5a
  

  jax authors

  committed
  92cdb5a

• [sparse] bcoo_dot_general_sampled: another special case

  jakevdp

  committed
  aad6a70

• Merge pull request jax-ml#14615 from skye:restore_opt_barrier

  Show description for 7177572
  

  jax authors

  committed
  7177572

• Merge pull request jax-ml#14653 from jakevdp:deprecate-array-methods

  Show description for b4c0146
  

  jax authors

  committed
  b4c0146

• Refactor Ref abstract type to contain other AbstractValues

  sharadmv

  committed
  4960e65

• Merge pull request jax-ml#14652 from sharadmv:flake-fix

  Show description for 8d0bdd2
  

  jax authors

  committed
  8d0bdd2

• Deprecate three jax.Array methods:

  Show description for a283aa0
  jakevdp

  committed
  a283aa0

• Fix nondeterminism issue with ordered effects

  sharadmv

  committed
  58c7e2e

Commits on Feb 23, 2023

• Add a helpful error message when device_putting with a Sharding that is incompatible with the shape of the input

  Show description for 5a8c12d
  

  yashk2810

  authored and

  jax authors

  committed
  5a8c12d

• fixes from reviewers

  mattjj

  committed
  c22da81

• add remat tutorial docs

  mattjj

  committed
  141996e

• Rollback of array fix again for perf regression.

  Show description for b502620
  

  pschuh

  authored and

  jax authors

  committed
  b502620

• Add support for XLAGather with 2-D Batch Dimensions for enable_xla=False

  Show description for 35a2735
  

  jax authors

  committed
  35a2735

• Merge pull request jax-ml#14631 from jakevdp:coo-padded

  Show description for 0bf960f
  

  jax authors

  committed
  0bf960f

• Merge pull request jax-ml#14632 from jakevdp:fix-bcoo-matmat

  Show description for 35f18fe
  

  jax authors

  committed
  35f18fe

• [sparse] fix bug in oob index correction

  jakevdp

  committed
  cff8eef

• [sparse] remove handling of padded indices from COO/CSR

  jakevdp

  committed
  bf1f5d2

• Merge pull request jax-ml#14630 from jakevdp:bcoo-dot-general-sampled

  Show description for 2d93b28
  

  jax authors

  committed
  2d93b28

• Merge pull request jax-ml#14598 from Tennessee-Wallaceh:Fix-student-t-sampling

  Show description for 81279e3
  

  jax authors

  committed
  81279e3

• Merge pull request jax-ml#14648 from jakevdp:is-ready-changelog

  Show description for c04fa2b
  

  jax authors

  committed
  c04fa2b

• DOC: add is_ready() to CHANGELOG

  jakevdp

  committed
  841bdce

• lax_scipy_test: Split into three so that each target is small enough to fit within a medium timeout.

  Show description for 09afbac
  

  pizzud

  authored and

  jax authors

  committed
  09afbac

• Merge pull request jax-ml#14638 from hawkinsp:conda

  Show description for a0f6354
  

  jax authors

  committed
  a0f6354

• Update student-t sampling to use correct key for gamma

  Tennessee-Wallaceh

  committed
  fbbdc35

Pagination

Previous

Next