Commits

Branch selector

 1ed6a81

User selector

All users

Datepicker

All time

Commit History

Commits on Jan 5, 2024

• Fix a type annotation typo.

  Show description for 1ed6a81
  

  zhangqiaorjc

  authored and

  jax authors

  committed
  1ed6a81

  Copy full SHA for 1ed6a81

• Update XLA dependency to use revision

  Show description for 059fdaf
  

  jax authors

  committed
  059fdaf

  Copy full SHA for 059fdaf

• [Pallas/Mosaic] Change Mosaic semaphores to be MemRef types

  Show description for d6a4723
  

  sharadmv

  authored and

  jax authors

  committed
  d6a4723

  Copy full SHA for d6a4723

Commits on Jan 4, 2024

• Merge pull request jax-ml#19201 from jakevdp:sort-kwargs

  Show description for 6dd5a69
  

  jax authors

  committed
  6dd5a69

  Copy full SHA for 6dd5a69

• Merge pull request jax-ml#19204 from jakevdp:doc-requirements

  Show description for 919233f
  

  jax authors

  committed
  919233f

  Copy full SHA for 919233f

• Introduce min entry size check for compilation cache.

  Show description for ea66029
  

  jax authors

  committed
  ea66029

  Copy full SHA for ea66029

• DOC: remove unnecessary doc build requirement

  jakevdp

  committed
  ccecdba

  Copy full SHA for ccecdba

• [array api] add stable & descending params to jnp.sort & jnp.argsort

  jakevdp

  committed
  8b62516

  Copy full SHA for 8b62516

• Fix/disable two tests failing in Windows CI.

  Show description for 8803774
  

  hawkinsp

  authored and

  jax authors

  committed
  8803774

  Copy full SHA for 8803774

• [XLA:Python] Raise an AttributeError if __cuda_array_interface__ is called on various invalid buffers, rather than a RuntimeError.

  Show description for 3ff4eb4
  

  hawkinsp

  authored and

  jax authors

  committed
  3ff4eb4

  Copy full SHA for 3ff4eb4

• Fix typo in pmap docstring

  Show description for ebc7af9
  

  tomcobley

  authored and

  jax authors

  committed
  ebc7af9

  Copy full SHA for ebc7af9

• jaxlib: avoid external build-time dependency on ml_dtypes

  Show description for 326d1d2
  

  Jake VanderPlas

  authored and

  jax authors

  committed
  326d1d2

  Copy full SHA for 326d1d2

• Guard call to os.register_at_fork() to unbreak Windows CI.

  Show description for 05a9906
  

  hawkinsp

  authored and

  jax authors

  committed
  05a9906

  Copy full SHA for 05a9906

• Raise rather than return NotImplementedError in PmapSharding.

  Show description for d56116c
  

  tomhennigan

  authored and

  jax authors

  committed
  d56116c

  Copy full SHA for d56116c

• [Jax] Allow to set the python traceback frames limit.

  Show description for 981b670
  

  jax authors

  committed
  981b670

  Copy full SHA for 981b670

• Rename batch_dims variable in shard_alike

  Show description for 7bba80d
  

  yashk2810

  authored and

  jax authors

  committed
  7bba80d

  Copy full SHA for 7bba80d

• [Pallas/GPU] Add support for jnp.cumsum

  Show description for 9112afb
  

  sharadmv

  authored and

  jax authors

  committed
  9112afb

  Copy full SHA for 9112afb

• Update XLA dependency to use revision

  Show description for 0d3fd53
  

  jax authors

  committed
  0d3fd53

  Copy full SHA for 0d3fd53

• Merge pull request jax-ml#19186 from jakevdp:asarray-copy

  Show description for c06c292
  

  jax authors

  committed
  c06c292

  Copy full SHA for c06c292

• Add correct batching rule for shard_alike

  Show description for 68abe0d
  

  yashk2810

  authored and

  jax authors

  committed
  68abe0d

  Copy full SHA for 68abe0d

• Add the githash that the jaxlib was built at to __init__.py. This is to allow identifying the githash of nightlies.

  Show description for 23b9c2a
  

  pschuh

  authored and

  jax authors

  committed
  23b9c2a

  Copy full SHA for 23b9c2a

Commits on Jan 3, 2024

• [array API] support copy argument to jnp.asarray

  jakevdp

  committed
  97fc213

  Copy full SHA for 97fc213

• Merge pull request jax-ml#19183 from jakevdp:dlpack-warning

  Show description for f1be301
  

  jax authors

  committed
  f1be301

  Copy full SHA for f1be301

• Merge pull request jax-ml#19182 from jakevdp:ndarray-item

  Show description for f8b7169
  

  jax authors

  committed
  f8b7169

  Copy full SHA for f8b7169

• DOC: add warning about dlpack and buffer mutation

  jakevdp

  committed
  df4e9c0

  Copy full SHA for df4e9c0

• jnp.ndarray.item(): add args support

  jakevdp

  committed
  47e5c81

  Copy full SHA for 47e5c81

• [Jax] Speedup stacks traceback

  Show description for b613679
  

  jax authors

  committed
  b613679

  Copy full SHA for b613679

• [Pallas/Mosaic] Add support for nested ref.at

  Show description for afa2f1e
  

  sharadmv

  authored and

  jax authors

  committed
  afa2f1e

  Copy full SHA for afa2f1e

• Update XLA dependency to use revision

  Show description for 7912e05
  

  jax authors

  committed
  7912e05

  Copy full SHA for 7912e05

• Error on conversion of empty arrays to boolean.

  Show description for c06e186
  

  Jake VanderPlas

  authored and

  jax authors

  committed
  c06e186

  Copy full SHA for c06e186

• Merge pull request jax-ml#19174 from mattjj:shmap-custom-vjp-replication-error-message

  Show description for 985a042
  

  jax authors

  committed
  985a042

  Copy full SHA for 985a042

Commits on Jan 2, 2024

• [Pallas] Refactor indexing primitives to use NDIndexer abstraction

  Show description for 836563f
  

  sharadmv

  authored and

  jax authors

  committed
  836563f

  Copy full SHA for 836563f

• Merge pull request jax-ml#19135 from eukub:concatenation-to-fstrings

  Show description for 8c5e7b2
  

  jax authors

  committed
  8c5e7b2

  Copy full SHA for 8c5e7b2

• [shard-map] improve error message when a custom_vjp bwd has extra psum

  mattjj

  committed
  12e57de

  Copy full SHA for 12e57de

• Delete sharding spec to HloSharding conversion since it's not used anymore.

  Show description for c0d4653
  

  yashk2810

  authored and

  jax authors

  committed
  c0d4653

  Copy full SHA for c0d4653

Pagination

Previous

Next