Commits

Breadcrumbs

History for

onmain

User selector

All users

Datepicker

All time

Commit History

Commits on Jun 18, 2022

• [new-remat] add _scan_partial_eval_custom rule for new remat

  Show description for 83a8dc4
  

  mattjj

  and

  sharadmv

  committed
  83a8dc4

  Copy full SHA for 83a8dc4

• Merge pull request jax-ml#11158 from jakevdp:x64-batching-test

  Show description for 9b7cad1
  

  jax authors

  committed
  9b7cad1

  Copy full SHA for 9b7cad1

• Merge pull request jax-ml#11150 from jakevdp:x64-scipy-sparse

  Show description for 812ff25
  

  jax authors

  committed
  812ff25

  Copy full SHA for 812ff25

Commits on Jun 17, 2022

• Merge pull request jax-ml#11162 from jakevdp:x64-sharded-jit

  Show description for 76387ae
  

  jax authors

  committed
  76387ae

  Copy full SHA for 76387ae

• Merge pull request jax-ml#11159 from jakevdp:x64-pjit-test

  Show description for 5b3ed30
  

  jax authors

  committed
  5b3ed30

  Copy full SHA for 5b3ed30

• [x64] make sharded_jit_test compatible with strict dtype promotion

  jakevdp

  committed
  1b2743d

  Copy full SHA for 1b2743d

• [x64] make pjit_test compatible with strict dtype promotion

  jakevdp

  committed
  51da9eb

  Copy full SHA for 51da9eb

• Merge pull request jax-ml#11157 from jakevdp:x64-vmap-test

  Show description for cc9ffbc
  

  jax authors

  committed
  cc9ffbc

  Copy full SHA for cc9ffbc

• [x64] make batching_test compatible with strict dtype promotion

  jakevdp

  committed
  9edbf9b

  Copy full SHA for 9edbf9b

• Merge pull request jax-ml#11155 from jakevdp:x64-ode-test

  Show description for 5236140
  

  jax authors

  committed
  5236140

  Copy full SHA for 5236140

• [x64] make vmap_test compatible with strict dtype promotion

  jakevdp

  committed
  7d21d6e

  Copy full SHA for 7d21d6e

• Merge pull request jax-ml#11151 from mattjj:input-fwd-residual-optimization

  Show description for 5318df6
  

  jax authors

  committed
  5318df6

  Copy full SHA for 5318df6

• optimize grad-of-jit not to pass input-residuals as intermediate-residuals

  Show description for 72a6790
  

  3 peoplecommitted
  72a6790

  Copy full SHA for 72a6790

• Merge pull request jax-ml#11116 from mattjj:djax-typecheck

  Show description for 5f849d3
  

  jax authors

  committed
  5f849d3

  Copy full SHA for 5f849d3

• [x64] make ode_test pass with strict dtype promotion

  jakevdp

  committed
  5e97011

  Copy full SHA for 5e97011

• [dynamic-shapes] initial support for dynamic shape typechecks

  Show description for f680269
  

  mattjj

  and

  dougalm

  committed
  f680269

  Copy full SHA for f680269

• [x64] make lax_control_flow_test compatible with strict dtype promotion

  jakevdp

  committed
  afe0dc1

  Copy full SHA for afe0dc1

• [x64] make jax.scipy.sparse.linalg compatible with strict dtype promotion

  jakevdp

  committed
  4c4a83b

  Copy full SHA for 4c4a83b

• [JAX] Drop the private _process_pytree method from tree_util.

  Show description for 7782fee
  

  hawkinsp

  authored and

  jax authors

  committed
  7782fee

  Copy full SHA for 7782fee

• Add __repr__ to Array. It works exactly as it does for DA and SDA when it is fully addressable. Otherwise it works like GDA.

  Show description for b2e1d81
  

  yashk2810

  authored and

  jax authors

  committed
  b2e1d81

  Copy full SHA for b2e1d81

• [x64] make checkify compatible with strict dtype promotion

  jakevdp

  committed
  305f5e0

  Copy full SHA for 305f5e0

• Merge pull request jax-ml#11137 from jakevdp:line-search-bug

  Show description for 21002b5
  

  jax authors

  committed
  21002b5

  Copy full SHA for 21002b5

• First CL to integrate jax.Array into pmap.

  Show description for 6ed94ef
  

  yashk2810

  authored and

  jax authors

  committed
  6ed94ef

  Copy full SHA for 6ed94ef

Commits on Jun 16, 2022

• Merge pull request jax-ml#11104 from jakevdp:x64-scipy-ndimage-test

  Show description for dd4a169
  

  jax authors

  committed
  dd4a169

  Copy full SHA for dd4a169

• Merge pull request jax-ml#11135 from jakevdp:x64-host-callback-test

  Show description for 6c8cf80
  

  jax authors

  committed
  6c8cf80

  Copy full SHA for 6c8cf80

• [x64] make scipy_ndimage_test compatible with strict dtype promotion

  jakevdp

  committed
  8b642b2

  Copy full SHA for 8b642b2

• Merge pull request jax-ml#11136 from jakevdp:x64-pmap-test

  Show description for e426d6b
  

  jax authors

  committed
  e426d6b

  Copy full SHA for e426d6b

• BUG: fix incorrect order of operations in line search

  jakevdp

  committed
  1567adf

  Copy full SHA for 1567adf

• Merge pull request jax-ml#11124 from jakevdp:x64-svd-test

  Show description for d58ba61
  

  jax authors

  committed
  d58ba61

  Copy full SHA for d58ba61

• Merge pull request jax-ml#11134 from jakevdp:x64-jaxpr-effects-test

  Show description for 7df0bbb
  

  jax authors

  committed
  7df0bbb

  Copy full SHA for 7df0bbb

• Merge pull request jax-ml#11132 from jakevdp:x64-lax-numpy-vectorize-test

  Show description for dc23e83
  

  jax authors

  committed
  dc23e83

  Copy full SHA for dc23e83

• [x64] make pmap_test compatible with strict dtype promotion

  jakevdp

  committed
  018e795

  Copy full SHA for 018e795

• Merge pull request jax-ml#11131 from jakevdp:x64-lax-numpy-indexing-test

  Show description for e66a565
  

  jax authors

  committed
  e66a565

  Copy full SHA for e66a565

• [x64] make host_callback_test compatible with strict dtype promotion

  jakevdp

  committed
  23510d0

  Copy full SHA for 23510d0

• [x64] make jaxpr_effects_test pass with strict dtype promotion

  jakevdp

  committed
  16ad959

  Copy full SHA for 16ad959

Pagination

Previous

Next