Commits

Branch selector

 b0621e3

User selector

All users

Datepicker

All time

Commit History

Commits on Nov 3, 2022

• Fix the vmap rule for remat_p

  Show description for b0621e3
  

  apaszke

  authored and

  jax authors

  committed
  b0621e3

  Copy full SHA for b0621e3

• Merge pull request jax-ml#13080 from jakevdp:fix-jet

  Show description for c8767ff
  

  jax authors

  committed
  c8767ff

  Copy full SHA for c8767ff

• BUG: fix jet rule for dynamic_slice

  jakevdp

  committed
  243b931

  Copy full SHA for 243b931

• Merge pull request jax-ml#13078 from jakevdp:bcoo-indexing

  Show description for eb9ac0d
  

  jax authors

  committed
  eb9ac0d

  Copy full SHA for eb9ac0d

• Rename ReshapeableDevicesSharding to PositionalSharding and add an alias NamedSharding for MeshPspecSharding.

  Show description for cc5af7e
  

  yashk2810

  authored and

  jax authors

  committed
  cc5af7e

  Copy full SHA for cc5af7e

• Merge pull request jax-ml#13083 from jakevdp:jet-dynamic-update

  Show description for 5448ea6
  

  jax authors

  committed
  5448ea6

  Copy full SHA for 5448ea6

• Add c++ pickling for all the jax Sharding types.

  Show description for 8b81240
  

  pschuh

  authored and

  jax authors

  committed
  8b81240

  Copy full SHA for 8b81240

Commits on Nov 2, 2022

• Clarify the limitation of running jax2tf converted pjit models on CPUs/GPUs.

  Show description for d573ce1
  

  rxsang

  authored and

  jax authors

  committed
  d573ce1

  Copy full SHA for d573ce1

• Enable state effect in cond_p (except in grad and vmap)

  Show description for e1af93a
  

  sharadmv

  authored and

  jax authors

  committed
  e1af93a

  Copy full SHA for e1af93a

• add jet rule for dynamic_update_slice

  jakevdp

  committed
  db7eea1

  Copy full SHA for db7eea1

• Increase the shard count of scipy_stats_test because it is timing out in OSS builds.

  Show description for 2dc8043
  

  yashk2810

  authored and

  jax authors

  committed
  2dc8043

  Copy full SHA for 2dc8043

• [JAX] Add RunTimeError to host_local_array_to_global_array

  Show description for b467feb
  

  jax authors

  committed
  b467feb

  Copy full SHA for b467feb

• Test dynamic_arg_shardings only for '==' equality not also the default pointer equality. Also add tests which checks this behavior and makes sure that we don't fallback to python

  Show description for 27bb347
  

  yashk2810

  authored and

  jax authors

  committed
  27bb347

  Copy full SHA for 27bb347

• Merge pull request jax-ml#13039 from skye:cache_compile_time_heuristic

  Show description for ef63f75
  

  jax authors

  committed
  ef63f75

  Copy full SHA for ef63f75

• [sparse] simplify BCOO indexing implementation

  jakevdp

  committed
  6ed9b14

  Copy full SHA for 6ed9b14

• Don't assume that vmap doesn't introduce constants

  Show description for 94ba43b
  

  apaszke

  authored and

  jax authors

  committed
  94ba43b

  Copy full SHA for 94ba43b

• Update docker images to add python 3.11 support.

  Show description for d77cccf
  

  yashk2810

  authored and

  jax authors

  committed
  d77cccf

  Copy full SHA for d77cccf

• Merge pull request jax-ml#13065 from sharadmv:vis-colors

  Show description for 93bcb59
  

  jax authors

  committed
  93bcb59

  Copy full SHA for 93bcb59

• Add new config jax_persistent_cache_min_compile_time_secs.

  Show description for cc51710
  skye

  committed
  cc51710

  Copy full SHA for cc51710

Commits on Nov 1, 2022

• Add colors to sharding visualization

  sharadmv

  committed
  3bbd5f3

  Copy full SHA for 3bbd5f3

• Merge pull request jax-ml#13015 from jurahul:master

  Show description for 60b9746
  

  jax authors

  committed
  60b9746

  Copy full SHA for 60b9746

• Raise an error if jax_array is not enabled when use jax.device_put with a Sharding as input.

  Show description for e881d16
  

  yashk2810

  authored and

  jax authors

  committed
  e881d16

  Copy full SHA for e881d16

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

  Show description for ef0f64e
  

  tlu7

  authored and

  jax authors

  committed
  ef0f64e

  Copy full SHA for ef0f64e

• Make device_put accept a prefix tree with Sharding leaves as the second argument

  Show description for bb07028
  

  hyeontaek

  authored and

  jax authors

  committed
  bb07028

  Copy full SHA for bb07028

• [sparse] fix a typo in n_sparse calculation.

  Show description for d207194
  

  tlu7

  authored and

  jax authors

  committed
  d207194

  Copy full SHA for d207194

• Fix docstring in GDA

  Show description for 52d29e1
  

  prazek

  authored and

  jax authors

  committed
  52d29e1

  Copy full SHA for 52d29e1

• Create a table of contents for jax2tf documentation.

  Show description for d08130b
  

  rxsang

  authored and

  jax authors

  committed
  d08130b

  Copy full SHA for d08130b

• Skip checkify checks for pjrt C API because of a failing pjit test with jax.Array is switched on.

  Show description for 08fbbc3
  

  yashk2810

  authored and

  jax authors

  committed
  08fbbc3

  Copy full SHA for 08fbbc3

Commits on Oct 31, 2022

• Add lax.cumlogsumexp for cumulative logsumexp operations.

  Show description for 5adfb08
  

  srvasude

  authored and

  jax authors

  committed
  5adfb08

  Copy full SHA for 5adfb08

• Rollback of:

  Show description for 06c1d8e
  

  Jake VanderPlas

  authored and

  jax authors

  committed
  06c1d8e

  Copy full SHA for 06c1d8e

• 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…

  Show description for ca1f58e
  

  yashk2810

  authored and

  jax authors

  committed
  ca1f58e

  Copy full SHA for ca1f58e

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

  Show description for f3ddd56
  

  jax authors

  committed
  f3ddd56

  Copy full SHA for f3ddd56

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

  Show description for 71edfc7
  

  jax authors

  committed
  71edfc7

  Copy full SHA for 71edfc7

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

  jakevdp

  committed
  2416d15

  Copy full SHA for 2416d15

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

  Show description for 32a0ea8
  

  yashk2810

  authored and

  jax authors

  committed
  32a0ea8

  Copy full SHA for 32a0ea8

Pagination

Previous

Next