Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Apr 29, 2023

• Add new attribute `function_list` to XLACallModule and bump the version.

  Show description for 63d87c6
  

  christisg

  authored and

  jax authors

  committed
  63d87c6

  Copy full SHA for 63d87c6

• [jax2tf] Simplify back_compat_test.py to use jax_export mechanisms to run

  Show description for 5b7e8d0
  

  gnecula

  authored and

  jax authors

  committed
  5b7e8d0

  Copy full SHA for 5b7e8d0

Commits on Apr 28, 2023

• Disable tsan CI for random_test_with_custom_prng to avoid timeouts.

  Show description for c662fd2
  

  skye

  authored and

  jax authors

  committed
  c662fd2

  Copy full SHA for c662fd2

• Merge pull request #15790 from jakevdp:sparse-doc

  Show description for b8f3caf
  

  jax authors

  committed
  b8f3caf

  Copy full SHA for b8f3caf

• DOC: document jax.experimental.sparse.linalg

  jakevdp

  committed
  e059e3b

  Copy full SHA for e059e3b

• Merge pull request #15779 from jakevdp:keyty-itemsize

  Show description for 0814b87
  

  jax authors

  committed
  0814b87

  Copy full SHA for 0814b87

• Merge pull request #15770 from gnecula:clean_call_tf

  Show description for 566f175
  

  jax authors

  committed
  566f175

  Copy full SHA for 566f175

• [call_tf] Some cleanup of call_tf

  Show description for 161664e
  gnecula

  committed
  161664e

  Copy full SHA for 161664e

• Improve error message in _create_device_mesh_for_nd_torus

  Show description for 818805f
  

  jax authors

  committed
  818805f

  Copy full SHA for 818805f

• Add new attribute `function_list` to XLACallModule and bump the version.

  Show description for 5b4388a
  

  maxwillzq

  authored and

  jax authors

  committed
  5b4388a

  Copy full SHA for 5b4388a

Commits on Apr 27, 2023

• KeyArray: define itemsize on opaque dtype

  jakevdp

  committed
  054fca5

  Copy full SHA for 054fca5

• Revert: Switch to using Clang as the default compiler.

  Show description for 84c5169
  

  hawkinsp

  authored and

  jax authors

  committed
  84c5169

  Copy full SHA for 84c5169

• KeyArray: add size attribute

  jakevdp

  committed
  50405b1

  Copy full SHA for 50405b1

• Preserve the sharding type of physical sharding on logical sharding when `.sharding` is accessed on a PRNGKeyArray

  Show description for 86c1f5b
  

  yashk2810

  authored and

  jax authors

  committed
  86c1f5b

  Copy full SHA for 86c1f5b

• [shape_poly] Add support for shape polymorphism for _unsafe_rbg_split.

  Show description for 7d6fb53
  

  voutcn

  authored and

  jax authors

  committed
  7d6fb53

  Copy full SHA for 7d6fb53

• Merge pull request #15753 from gnecula:poly_unif

  Show description for 5027543
  

  jax authors

  committed
  5027543

  Copy full SHA for 5027543

• [shape_poly] Refactor the unification of the argument abstract values with the actual arguments

  Show description for 876c53a
  gnecula

  committed
  876c53a

  Copy full SHA for 876c53a

• Fix bug in the JAX spectral bisection eigensolver implementing the QDWH-eigh algorithm.

  Show description for c6457f3
  

  jax authors

  committed
  c6457f3

  Copy full SHA for c6457f3

Commits on Apr 26, 2023

• Default jax_spmd_mode to allow_jit which will allow explicit jax.jit to not raise the multihost error (since jit and pjit have been merged).

  Show description for 34d5a62
  

  yashk2810

  authored and

  jax authors

  committed
  34d5a62

  Copy full SHA for 34d5a62

• Increase sharding count on nn_test and svd_test to avoid ASAN timeouts.

  Show description for 67d80c2
  

  skye

  authored and

  jax authors

  committed
  67d80c2

  Copy full SHA for 67d80c2

• Merge pull request #15762 from jakevdp:keyarray-eq

  Show description for 139e1c2
  

  jax authors

  committed
  139e1c2

  Copy full SHA for 139e1c2

• KeyArray: use assertArraysEqual in place of assertKeysEqual

  jakevdp

  committed
  e46d7f6

  Copy full SHA for e46d7f6

• KeyArray: implement __eq__ and __ne__

  jakevdp

  committed
  fcffbac

  Copy full SHA for fcffbac

• Merge pull request #15757 from jakevdp:keyarray-operators

  Show description for be74b07
  

  jax authors

  committed
  be74b07

  Copy full SHA for be74b07

• KeyArray: better errors for operators

  jakevdp

  committed
  a47a71f

  Copy full SHA for a47a71f

• Merge pull request #15748 from jakevdp:keyarray-at

  Show description for a6b052a
  

  jax authors

  committed
  a6b052a

  Copy full SHA for a6b052a

• Convert inspect_sharding to register the handler directly in c++ so that it can

  Show description for 5f4408d
  

  pschuh

  authored and

  jax authors

  committed
  5f4408d

  Copy full SHA for 5f4408d

• Merge pull request #15755 from jakevdp:sharding-constraint-doc

  Show description for 1551d9e
  

  jax authors

  committed
  1551d9e

  Copy full SHA for 1551d9e

• [2/n] Embed the tf.Graph into the stablehlo.custom_call.

  Show description for af051ba
  

  maxwillzq

  authored and

  jax authors

  committed
  af051ba

  Copy full SHA for af051ba

• Document jax.lax.with_sharding_constraint

  jakevdp

  committed
  8dc06ed

  Copy full SHA for 8dc06ed

• Merge pull request #15733 from gnecula:export_import

  Show description for a51c8cc
  

  jax authors

  committed
  a51c8cc

  Copy full SHA for a51c8cc

• Use jax.jit since it will ignore any surrounding mesh context managers for sync_global_devices and broadcast_one_to_all

  Show description for 4a746b3
  

  yashk2810

  authored and

  jax authors

  committed
  4a746b3

  Copy full SHA for 4a746b3

• Merge pull request #15732 from gnecula:tf_fix_float0

  Show description for 4f67bd9
  

  jax authors

  committed
  4f67bd9

  Copy full SHA for 4f67bd9

Commits on Apr 25, 2023

• PRNGKeyArray: implement scatter/gather via .at()

  jakevdp

  committed
  6e84ed2

  Copy full SHA for 6e84ed2

• Merge pull request #15745 from jakevdp:xlogx-test

  Show description for 4e88943
  

  jax authors

  committed
  4e88943

  Copy full SHA for 4e88943

Pagination

Previous

Next