Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Nov 20, 2021

• Switch most uses of jax.lax in jax2tf to use the public API.

  Show description for fca3da5
  

  hawkinsp

  authored and

  jax authors

  committed
  fca3da5

  Copy full SHA for fca3da5

• Add local_data API for GSDA

  Show description for 65a99db
  

  yashk2810

  authored and

  jax authors

  committed
  65a99db

  Copy full SHA for 65a99db

Commits on Nov 19, 2021

• Adding support for a special value for in_axis_resources (pjit.FROM_GSDA) when GSDA is an input.

  Show description for d1de309
  

  yashk2810

  authored and

  jax authors

  committed
  d1de309

  Copy full SHA for d1de309

• Merge pull request jax-ml#8624 from jakevdp:quantile-complex

  Show description for 34a2ffc
  

  jax authors

  committed
  34a2ffc

  Copy full SHA for 34a2ffc

• Make it easier to see what are the difference between two structures.

  Show description for cad3fd8
  

  jax authors

  committed
  cad3fd8

  Copy full SHA for cad3fd8

• [sparse] specify operand layouts in cusparse.py

  Show description for a93c99d
  

  Jake VanderPlas

  authored and

  jax authors

  committed
  a93c99d

  Copy full SHA for a93c99d

• jnp.quantile: explicitly raise error for complex input

  jakevdp

  committed
  7227636

  Copy full SHA for 7227636

• Merge pull request jax-ml#8552 from mattjj:elide-more-convert-element-types

  Show description for f08a5a0
  

  jax authors

  committed
  f08a5a0

  Copy full SHA for f08a5a0

• generalize jaxpr simplification machinery

  Show description for abbf78b
  mattjj

  committed
  abbf78b

  Copy full SHA for abbf78b

• Merge pull request jax-ml#8606 from hawkinsp:laxsplit

  Show description for 75e063c
  

  jax authors

  committed
  75e063c

  Copy full SHA for 75e063c

• Merge pull request jax-ml#8576 from jakevdp:bcoo-dot-general-types

  Show description for 602dd39
  

  jax authors

  committed
  602dd39

  Copy full SHA for 602dd39

Commits on Nov 18, 2021

• [sparse] bcoo_[sp]dot_general: require inputs to have the same type

  jakevdp

  committed
  193746f

  Copy full SHA for 193746f

• Split windowed reductions and their gradients into a separate file inside the lax implementation.

  hawkinsp

  committed
  45d7ade

  Copy full SHA for 45d7ade

• Merge pull request jax-ml#8602 from hawkinsp:mypyjaxlib

  Show description for 4c504ea
  

  jax authors

  committed
  4c504ea

  Copy full SHA for 4c504ea

• Increment minimum jaxlib version in mypy precommit hook.

  hawkinsp

  committed
  58d5ee9

  Copy full SHA for 58d5ee9

• Merge pull request jax-ml#8565 from jakevdp:spdot-general

  Show description for f391b5b
  

  jax authors

  committed
  f391b5b

  Copy full SHA for f391b5b

• Merge pull request jax-ml#8601 from mattjj:fix-vmap-ppermute

  Show description for 3ee76a8
  

  jax authors

  committed
  3ee76a8

  Copy full SHA for 3ee76a8

• [sparse] spdot_general: implement many more cases

  jakevdp

  committed
  848675d

  Copy full SHA for 848675d

• Merge pull request jax-ml#8584 from jakevdp:fix-sum-duplicates

  Show description for d422554
  

  jax authors

  committed
  d422554

  Copy full SHA for d422554

• Merge pull request jax-ml#8598 from hawkinsp:jaxlib

  Show description for ef6fb07
  

  jax authors

  committed
  ef6fb07

  Copy full SHA for ef6fb07

• [sparse] fix batched version of BCOO.sum_duplicates

  jakevdp

  committed
  acca0bf

  Copy full SHA for acca0bf

• Merge pull request jax-ml#8567 from jakevdp:unique-fill-value

  Show description for fa5520b
  

  jax authors

  committed
  fa5520b

  Copy full SHA for fa5520b

• Merge pull request jax-ml#8581 from jakevdp:sparse-matmul-dtype

  Show description for bf74f2e
  

  jax authors

  committed
  bf74f2e

  Copy full SHA for bf74f2e

• make vmap ppermute consistent with pmap/docstring

  Show description for 2cb2358
  mattjj

  committed
  2cb2358

  Copy full SHA for 2cb2358

• Simplify implementation of lower_fun in XLA and MLIR lowering paths.

  Show description for 606ca26
  

  hawkinsp

  authored and

  jax authors

  committed
  606ca26

  Copy full SHA for 606ca26

• Increase minimum jaxlib version to 0.1.74.

  hawkinsp

  committed
  3fd3c46

  Copy full SHA for 3fd3c46

• [MLIR] Add support for pruning token values from computation inputs/outputs.

  Show description for 52421da
  

  hawkinsp

  authored and

  jax authors

  committed
  52421da

  Copy full SHA for 52421da

• Merge pull request jax-ml#8594 from hawkinsp:docs

  Show description for 2bd40e4
  

  jax authors

  committed
  2bd40e4

  Copy full SHA for 2bd40e4

• Delete the XLA in Python notebook.

  Show description for 58199b4
  hawkinsp

  committed
  58199b4

  Copy full SHA for 58199b4

• Merge pull request jax-ml#8516 from gnecula:tf_image_resize

  Show description for 639517f
  

  jax authors

  committed
  639517f

  Copy full SHA for 639517f

• Added workaround for bug in XLA

  gnecula

  committed
  3715fcb

  Copy full SHA for 3715fcb

• [shape_poly] Refactor arange and image_resize for shape polymorphism

  Show description for 75155f5
  gnecula

  committed
  75155f5

  Copy full SHA for 75155f5

• Merge pull request jax-ml#8574 from hawkinsp:jaxlib

  Show description for c37c62c
  

  jax authors

  committed
  c37c62c

  Copy full SHA for c37c62c

• Merge pull request jax-ml#8583 from jakevdp:remove-prints

  Show description for 21d4ca3
  

  jax authors

  committed
  21d4ca3

  Copy full SHA for 21d4ca3

• jnp.unique: ensure that output dtype is not affected by fill_value

  jakevdp

  committed
  0bee9b3

  Copy full SHA for 0bee9b3

Pagination

Previous

Next