Commits

Branch selector

 main

User selector

All users

All time

Commit History

Commits on Oct 31, 2022

• Add a new jax.spmd_mode config for preventing unintentional hangs and incorrect results when users pass jax.Arrays that span across multiple processes (i.e. not fully addressable) to jit or j…

  

  yashk2810

  authored and

  jax authors

  committedOct 31, 2022
  ca1f58e

• Merge pull request google#13035 from jakevdp:jnp-put

  

  jax authors

  committedOct 31, 2022
  f3ddd56

• Merge pull request google#12670 from jakevdp:numpy-check-arraylike

  

  jax authors

  committedOct 31, 2022
  71edfc7

• Call _check_arraylike for jnp.linalg & jnp.fft functions

  jakevdp

  committedOct 31, 2022
  2416d15

• Add global_shards to jax.Array as it exists on GDA and is being used in various places.

  

  yashk2810

  authored and

  jax authors

  committedOct 31, 2022
  32a0ea8

Commits on Oct 29, 2022

• Merge pull request google#13040 from froystig:rng-part-bitschange-extents

  

  jax authors

  committedOct 29, 2022
  5b0cb11

• integrate new (partitionable, count-space-exhaustive) counts generation

  mattjj

  committedOct 29, 2022
  213d2c8

• [XLA:CPU][XLA:GPU] Remove Sharding CustomCall when num_partitions==1, or return an error when num_partitions>1 but SPMD partitioning is not enabled.

  

  cezheng

  authored and

  jax authors

  committedOct 29, 2022
  ef4fd14

Commits on Oct 28, 2022

• Merge pull request google#13022 from mattjj:leak-checker-improvements

  

  jax authors

  committedOct 28, 2022
  8dea82e

• Merge pull request google#11077 from DanPuzzuoli:ode_dt_max

  

  jax authors

  committedOct 28, 2022
  9dac458

• Merge pull request google#13037 from mattjj:fix-ci-failure

  

  jax authors

  committedOct 28, 2022
  03b2691

• fix ci failure by skipping tests on gpu

  mattjj

  committedOct 28, 2022
  0203534

• wip bits-changing partitionable rng based on iota raveling

  

  froystig

  and

  mattjj

  committedOct 28, 2022
  63bfb87

• Point to ndarray.at from docstring of unimplemented jnp.put & jnp.place

  jakevdp

  committedOct 28, 2022
  8bde3a0

• jax.numpy docstrings: remove empty parameters section

  jakevdp

  committedOct 28, 2022
  8562e76

• make leak checker errors explain why objects are alive

  

  3 peoplecommittedOct 28, 2022
  6ebf44a

• Merge pull request google#13010 from jakevdp:annotate-lax-numpy

  

  jax authors

  committedOct 28, 2022
  f9e7629

• Merge pull request google#13012 from mattjj:rng-part-overgenerate

  

  jax authors

  committedOct 28, 2022
  89b240b

• Improve effect support on internal backends.

  

  hyeontaek

  authored and

  jax authors

  committedOct 28, 2022
  54967e9

• [typing] annotate next section of lax_numpy.py

  jakevdp

  committedOct 28, 2022
  15b489f

• partitionable threefry PRNG random bits implementation

  

  froystig

  and

  mattjj

  committedOct 28, 2022
  c8b9280

• Add the full shape to the error message too (in pjit_check_aval_sharding) to give users full information about what is going on.

  

  yashk2810

  authored and

  jax authors

  committedOct 28, 2022
  2cc4fd2

• Add a warning to device.live_buffers() as it is going to be deprecated with jax.Array and instruct users to use jax.live_arrays() instead.

  

  cky9301

  authored and

  jax authors

  committedOct 28, 2022
  57eb19f

• Merge pull request google#13023 from jakevdp:jnp-window

  

  jax authors

  committedOct 28, 2022
  1816263

• Merge pull request google#12989 from apaszke:xmap-better-axis-checks

  

  jax authors

  committedOct 28, 2022
  42e75a2

• [jax2tf] Improve handling of XlaSharding ops.

  

  gnecula

  authored and

  jax authors

  committedOct 28, 2022
  c1c22d9

• Use d.process_index == d.client.process_index() to find addressable devices from a list of addressable and non-addressable devices.

  

  yashk2810

  authored and

  jax authors

  committedOct 28, 2022
  4a21d41

Commits on Oct 27, 2022

• Remove error-prone most_recent_entry() support from lu.cache.

  

  pschuh

  authored and

  jax authors

  committedOct 27, 2022
  5cfc708

• jax.numpy: implement window functions in terms of lax ops

  jakevdp

  committedOct 27, 2022
  0a79fa4

• jax.numpy: implement window functions in terms of lax ops

  jakevdp

  committedOct 27, 2022
  51242bc

• [sparse] Update the default CUSPARSE SPMV and SPMM algorithms in jaxlib.

  

  tlu7

  authored and

  jax authors

  committedOct 27, 2022
  66e75ed

• Internal visibility change

  

  hyeontaek

  authored and

  jax authors

  committedOct 27, 2022
  fc8f40c

• Merge pull request google#13013 from ROCmSoftwarePlatform:rocm_update_amdgpu_archs

  

  jax authors

  committedOct 27, 2022
  0c119cb

• Internal change

  

  Jake VanderPlas

  authored and

  jax authors

  committedOct 27, 2022
  f699acd

• Update debugging_primitives_test to not use nontrivial floating point text comparisons

  

  sharadmv

  authored and

  jax authors

  committedOct 27, 2022
  3e38675

Pagination

PreviousNext