Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Feb 15, 2023

• migrate more internal dependencies from `jax.core` to `jax._src.core`

  Show description for cb8dcce
  

  froystig

  authored and

  jax authors

  committed
  cb8dcce

  Copy full SHA for cb8dcce

• Add clear_cache endpoint to python pjit and cpp pjit functions.

  Show description for b476661
  

  yashk2810

  authored and

  jax authors

  committed
  b476661

  Copy full SHA for b476661

Commits on Feb 14, 2023

• Merge pull request #14446 from jakevdp:initial-scalar

  Show description for e74852f
  

  jax authors

  committed
  e74852f

  Copy full SHA for e74852f

• jax.numpy reductions: require initial to be a scalar

  Show description for dafb88a
  jakevdp

  committed
  dafb88a

  Copy full SHA for dafb88a

• Merge pull request #14474 from jakevdp:doc-array-methods

  Show description for c2b7c5f
  

  jax authors

  committed
  c2b7c5f

  Copy full SHA for c2b7c5f

• Fix issue where HLO could not be generated for custom gradient.

  Show description for 59e9746
  

  k-w-w

  authored and

  jax authors

  committed
  59e9746

  Copy full SHA for 59e9746

• Merge pull request #14472 from nouiz:shmap_jep_fixes

  Show description for a9ef989
  

  jax authors

  committed
  a9ef989

  Copy full SHA for a9ef989

• DOC: improve documentation for jax.Array methods

  jakevdp

  committed
  5958bf0

  Copy full SHA for 5958bf0

• DOC: improve documentation for jax.Array methods

  jakevdp

  committed
  967f211

  Copy full SHA for 967f211

• Merge pull request #14453 from jakevdp:dtypes-doc

  Show description for 5860cfd
  

  jax authors

  committed
  5860cfd

  Copy full SHA for 5860cfd

• Opt into higher matmul precision for A100 and TPU tests.

  Show description for 33bed1e
  

  hawkinsp

  authored and

  jax authors

  committed
  33bed1e

  Copy full SHA for 33bed1e

• Merge pull request #14275 from xoiga123:fix-jax.numpy.hsplit

  Show description for aa98c99
  

  jax authors

  committed
  aa98c99

  Copy full SHA for aa98c99

• Use right fct name.

  nouiz

  committed
  93c9313

  Copy full SHA for 93c9313

• Be consistent in the index used

  nouiz

  committed
  d2bb1e0

  Copy full SHA for d2bb1e0

• DOC: add docs for jax.dtypes module

  jakevdp

  committed
  11e3219

  Copy full SHA for 11e3219

• Merge pull request #14456 from jakevdp:jax-typing-public

  Show description for 47dca67
  

  jax authors

  committed
  47dca67

  Copy full SHA for 47dca67

• Small crash fixes

  nouiz

  committed
  6735102

  Copy full SHA for 6735102

• License cleanup.

  Show description for 658a934
  

  hawkinsp

  authored and

  jax authors

  committed
  658a934

  Copy full SHA for 658a934

• Catch the NaN's and raise a better error message when jax_debug_nans flag is True.

  Show description for 1c651f2
  

  yashk2810

  authored and

  jax authors

  committed
  1c651f2

  Copy full SHA for 1c651f2

• [JAX] Improve error message when jit tracer passed to a shape.

  Show description for 995ef40
  

  zeynepCankara

  authored and

  jax authors

  committed
  995ef40

  Copy full SHA for 995ef40

• clean up no-need exception raise.

  Show description for 4237f22
  

  maxwillzq

  authored and

  jax authors

  committed
  4237f22

  Copy full SHA for 4237f22

• [sparse] enable bcsr_dot_general cusparse lowering

  Show description for 15196bc
  

  Jake VanderPlas

  authored and

  jax authors

  committed
  15196bc

  Copy full SHA for 15196bc

• Implement lowering for convolutions with dynamic padding

  Show description for 582c042
  

  gnecula

  authored and

  jax authors

  committed
  582c042

  Copy full SHA for 582c042

• Bump minimum Python version in "contributing" docs.

  Show description for 0f7ffb8
  

  hawkinsp

  authored and

  jax authors

  committed
  0f7ffb8

  Copy full SHA for 0f7ffb8

• Merge pull request #14457 from mattjj:djax-bug-fix

  Show description for 9e01ee4
  

  jax authors

  committed
  9e01ee4

  Copy full SHA for 9e01ee4

• Fix xmap staging rule to handle positional semantics

  Show description for 442aa02
  

  sharadmv

  authored and

  jax authors

  committed
  442aa02

  Copy full SHA for 442aa02

Commits on Feb 13, 2023

• Make colab_gpu.ipynb compatible with newer JAX versions

  Show description for e1ff0c1
  

  Jake VanderPlas

  authored and

  jax authors

  committed
  e1ff0c1

  Copy full SHA for e1ff0c1

• Expose jax.typing & update docs

  jakevdp

  committed
  7975192

  Copy full SHA for 7975192

• fix minor broadcasting bug

  Show description for 96c558d
  

  mattjj

  and

  apaszke

  committed
  96c558d

  Copy full SHA for 96c558d

• Plumb spmd_axis_name through batch_jaxpr2 and batch_jaxpr

  Show description for d0eedf7
  

  yashk2810

  authored and

  jax authors

  committed
  d0eedf7

  Copy full SHA for d0eedf7

• Minimize exported names from jax.experimental.maps.

  Show description for 4a523e3
  

  hawkinsp

  authored and

  jax authors

  committed
  4a523e3

  Copy full SHA for 4a523e3

• Change the `axis_resources` argument of `with_sharding_constraint` to `shardings` to match `pjit` and `jit`.

  Show description for 2fc64be
  

  yashk2810

  authored and

  jax authors

  committed
  2fc64be

  Copy full SHA for 2fc64be

• Merge pull request #14365 from jakevdp:reducers-initial

  Show description for c49af18
  

  jax authors

  committed
  c49af18

  Copy full SHA for c49af18

• Merge pull request #14444 from jakevdp:fix-csr-lowering

  Show description for 83b7ba2
  

  jax authors

  committed
  83b7ba2

  Copy full SHA for 83b7ba2

• jax.numpy reductions: better validation of initial value

  jakevdp

  committed
  58323d5

  Copy full SHA for 58323d5

Pagination

Previous

Next