Commits

Branch selector

 655ecb4

User selector

All users

Datepicker

All time

Commit History

Commits on Aug 24, 2022

• fix bug issue template yaml parsing error

  froystig

  authored
  655ecb4

  Copy full SHA for 655ecb4

• Allow to_py() on Array if the sharding is fully replicated.

  Show description for 9736fac
  

  yashk2810

  authored and

  jax authors

  committed
  9736fac

  Copy full SHA for 9736fac

Commits on Aug 23, 2022

• Suppress msan failure in PRNG code.

  Show description for 160a6c5
  

  hawkinsp

  authored and

  jax authors

  committed
  160a6c5

  Copy full SHA for 160a6c5

• Merge pull request jax-ml#11944 from jakevdp:rm-tile

  Show description for 0a51a5a
  

  jax authors

  committed
  0a51a5a

  Copy full SHA for 0a51a5a

• Merge pull request jax-ml#12041 from jakevdp:fix-pow-jvp

  Show description for 2bdfdef
  

  jax authors

  committed
  2bdfdef

  Copy full SHA for 2bdfdef

• Merge pull request jax-ml#12035 from marcvanzee:patch-4

  Show description for 668d08f
  

  jax authors

  committed
  668d08f

  Copy full SHA for 668d08f

• Merge pull request jax-ml#12054 from jakevdp:lax-output

  Show description for b32701c
  

  jax authors

  committed
  b32701c

  Copy full SHA for b32701c

• Fix JVP rule for lax.pow()

  jakevdp

  committed
  b8fe0ab

  Copy full SHA for b8fe0ab

• Merge pull request jax-ml#12032 from ROCmSoftwarePlatform:rocm_hipsparse_fix

  Show description for 560cc36
  

  jax authors

  committed
  560cc36

  Copy full SHA for 560cc36

• lax.squeeze: ensure DeviceArray is returned

  jakevdp

  committed
  ab4ec58

  Copy full SHA for ab4ec58

• Shut down preemption sync manager if enabled.

  Show description for a73a6a8
  

  jax authors

  committed
  a73a6a8

  Copy full SHA for a73a6a8

• Enable fast path in the Array constructor. This means that the rearranging of _arrays according to the device_assignment won't happen when fastpath is enabled because we assume that jax transform…

  Show description for e8ec454
  

  yashk2810

  authored and

  jax authors

  committed
  e8ec454

  Copy full SHA for e8ec454

• [JAX] Prepare not to export jax._src by default.

  Show description for 335b2cf
  

  hawkinsp

  authored and

  jax authors

  committed
  335b2cf

  Copy full SHA for 335b2cf

• Merge pull request jax-ml#12046 from google:bug-report-template-title

  Show description for fd79974
  

  jax authors

  committed
  fd79974

  Copy full SHA for fd79974

• Remove deprecated array.tile() method

  jakevdp

  committed
  8378d08

  Copy full SHA for 8378d08

Commits on Aug 22, 2022

• remove 'BUG' title prefix from bug report template

  Show description for 825cbac
  froystig

  authored
  825cbac

  Copy full SHA for 825cbac

• Merge pull request jax-ml#12021 from jakevdp:copy-padding

  Show description for 3aa4277
  

  jax authors

  committed
  3aa4277

  Copy full SHA for 3aa4277

• Merge pull request jax-ml#12043 from google:yashk2810-patch-15

  Show description for 843920b
  

  jax authors

  committed
  843920b

  Copy full SHA for 843920b

• roll-forward jax-ml#11952, take 2

  Show description for 6071a8f
  

  froystig

  authored and

  jax authors

  committed
  6071a8f

  Copy full SHA for 6071a8f

• Add checkpointing support for Array similar to GDA.

  Show description for 7cdb7e1
  

  yashk2810

  authored and

  jax authors

  committed
  7cdb7e1

  Copy full SHA for 7cdb7e1

• Add upgrade=True to jax_array flag so that its marked as transient flag which will eventually be set to True.

  yashk2810

  authored
  37089ec

  Copy full SHA for 37089ec

• Merge pull request jax-ml#12036 from hawkinsp:gpu

  Show description for 384776f
  

  jax authors

  committed
  384776f

  Copy full SHA for 384776f

• Increase some test shardings to reduce CI timeouts under asan/tsan/msan.

  Show description for da4e79a
  

  hawkinsp

  authored and

  jax authors

  committed
  da4e79a

  Copy full SHA for da4e79a

• Relax test tolerance to fix GPU CI failure.

  hawkinsp

  committed
  c247b9b

  Copy full SHA for c247b9b

• Update no_xla_limitations.md

  marcvanzee

  authored
  4f78a1d

  Copy full SHA for 4f78a1d

Commits on Aug 21, 2022

• [ROCm] Enhance hipsparse to reach parity with cusparse based on commit jax-ml@d37b711.

  rsanthanam-amd

  committed
  82adc6a

  Copy full SHA for 82adc6a

Commits on Aug 20, 2022

• Make eager pmap tests pass with Array. Also add a slow path for Array in pmap similar to what SDA has. This is required for eager pmap. Adding a slow path removes the need for doing sharding ch…

  Show description for f905d98
  

  yashk2810

  authored and

  jax authors

  committed
  f905d98

  Copy full SHA for f905d98

• roll-forward jax-ml#11952

  Show description for 3a2f25f
  

  jax authors

  committed
  3a2f25f

  Copy full SHA for 3a2f25f

• roll-forward jax-ml#11952

  Show description for 9789e83
  

  froystig

  authored and

  jax authors

  committed
  9789e83

  Copy full SHA for 9789e83

• Add config.jax_array to _trace_context so that in can be used in lu.cache key.

  Show description for 78cfbeb
  

  yashk2810

  authored and

  jax authors

  committed
  78cfbeb

  Copy full SHA for 78cfbeb

• Merge pull request jax-ml#12022 from jakevdp:copy-sparse

  Show description for c7203df
  

  jax authors

  committed
  c7203df

  Copy full SHA for c7203df

Commits on Aug 19, 2022

• Merge pull request jax-ml#12013 from jakevdp:monoid-reducer

  Show description for 883fe88
  

  jax authors

  committed
  883fe88

  Copy full SHA for 883fe88

• [sparse] add trivial sparse rule for lax.copy_p

  jakevdp

  committed
  d78b969

  Copy full SHA for d78b969

• Merge pull request jax-ml#12014 from sudhakarsingh27:use_nightly_gpu_jaxlib_for_testing

  Show description for bfef22a
  

  jax authors

  committed
  bfef22a

  Copy full SHA for bfef22a

• Merge pull request jax-ml#12019 from jakevdp:export-copy-p

  Show description for 306a032
  

  jax authors

  committed
  306a032

  Copy full SHA for 306a032

Pagination

Previous

Next