Commits

Branch selector

 69969ef

User selector

All users

Datepicker

All time

Commit History

Commits on Mar 21, 2022

• add random.loggamma and improve dirichlet & beta implementation

  jakevdp

  committed
  69969ef

  Copy full SHA for 69969ef

Commits on Mar 19, 2022

• Merge pull request jax-ml#9964 from jakevdp:bcoo-mul

  Show description for 1ffa285
  

  jax authors

  committed
  1ffa285

  Copy full SHA for 1ffa285

• Set aval when creating the local_shards

  Show description for 1b17b54
  

  yashk2810

  authored and

  jax authors

  committed
  1b17b54

  Copy full SHA for 1b17b54

Commits on Mar 18, 2022

• [sparse] bcoo_mul: support mixing batch & sparse dims

  jakevdp

  committed
  b2e0dea

  Copy full SHA for b2e0dea

• Merge pull request jax-ml#9738 from jakevdp:refactor-reductions

  Show description for e392af3
  

  jax authors

  committed
  e392af3

  Copy full SHA for e392af3

• Merge pull request jax-ml#9962 from google:update-pypi

  Show description for b311730
  

  jax authors

  committed
  b311730

  Copy full SHA for b311730

• update version and changelog for pypi

  mattjj

  committed
  4c5d8e9

  Copy full SHA for 4c5d8e9

• Merge pull request jax-ml#9959 from mattjj:fix-9923

  Show description for b4f47c4
  

  jax authors

  committed
  b4f47c4

  Copy full SHA for b4f47c4

• Factor-out reductions from lax_numpy.py

  jakevdp

  committed
  121d8d6

  Copy full SHA for 121d8d6

• fix typo in jax-ml#9923

  mattjj

  committed
  d60d5d7

  Copy full SHA for d60d5d7

• Merge pull request jax-ml#9955 from nicholasjng:add-itemsize

  Show description for e9f59ae
  

  jax authors

  committed
  e9f59ae

  Copy full SHA for e9f59ae

• Add itemsize property to JAX arrays

  Show description for 9e149bb
  nicholasjng

  committed
  9e149bb

  Copy full SHA for 9e149bb

• Replace (deprecated) StrEnumAttr with EnumAttr.

  Show description for 6cd9804
  

  sdasgup3

  authored and

  jax authors

  committed
  6cd9804

  Copy full SHA for 6cd9804

• Merge pull request jax-ml#9950 from froystig:wrapped

  Show description for 1f95273
  

  jax authors

  committed
  1f95273

  Copy full SHA for 1f95273

• Internal change

  Show description for 4b81311
  

  yashk2810

  authored and

  jax authors

  committed
  4b81311

  Copy full SHA for 4b81311

• introduce a protocol for compilation wrappers

  froystig

  committed
  e04ea89

  Copy full SHA for e04ea89

• Merge pull request jax-ml#9948 from froystig:xmap-api-boundary

  Show description for f73a488
  

  jax authors

  committed
  f73a488

  Copy full SHA for f73a488

Commits on Mar 17, 2022

• mark xmap as an API boundary; filter experimental.maps from tracebacks and source info

  froystig

  committed
  a6841a9

  Copy full SHA for a6841a9

• Merge pull request jax-ml#9947 from google:update-pypi

  Show description for ce4ba11
  

  jax authors

  committed
  ce4ba11

  Copy full SHA for ce4ba11

• update version and changelog for pypi

  mattjj

  committed
  d2b393b

  Copy full SHA for d2b393b

• Add small and big matmul to api_benchmarks.

  Show description for 5d7f639
  

  zhangqiaorjc

  authored and

  jax authors

  committed
  5d7f639

  Copy full SHA for 5d7f639

• Merge pull request jax-ml#9942 from jakevdp:glu-fix

  Show description for 53f52cb
  

  jax authors

  committed
  53f52cb

  Copy full SHA for 53f52cb

• Merge pull request jax-ml#9944 from jakevdp:poly-refactor

  Show description for ae631e9
  

  jax authors

  committed
  ae631e9

  Copy full SHA for ae631e9

• lax_numpy: move poly functions into numpy.polynomial

  jakevdp

  committed
  603bb3c

  Copy full SHA for 603bb3c

• Merge pull request jax-ml#9943 from jakevdp:ufunc-imports

  Show description for 2d79a64
  

  jax authors

  committed
  2d79a64

  Copy full SHA for 2d79a64

• lax_numpy refactoring: directly import ufuncs

  jakevdp

  committed
  131c005

  Copy full SHA for 131c005

• Merge pull request jax-ml#9925 from jakevdp:refactor-index-ops

  Show description for d7bd93f
  

  jax authors

  committed
  d7bd93f

  Copy full SHA for d7bd93f

• jax.nn.glu: fix static argname issue

  jakevdp

  committed
  c762e07

  Copy full SHA for c762e07

• Revert previous change

  Show description for c3a4a6e
  

  tkoeppe

  authored and

  jax authors

  committed
  c3a4a6e

  Copy full SHA for c3a4a6e

• lax_numpy: factor out indexing tricks

  jakevdp

  committed
  0a72adb

  Copy full SHA for 0a72adb

• Merge pull request jax-ml#9938 from jakevdp:doc-disable-jit

  Show description for 250ef01
  

  jax authors

  committed
  250ef01

  Copy full SHA for 250ef01

• DOC: more info on disable_jit

  jakevdp

  committed
  625a69d

  Copy full SHA for 625a69d

• Reshape top_k operand to 2D by collapsing the batch dimensions when lowering.

  Show description for 1d5833d
  

  LenaMartens

  authored and

  jax authors

  committed
  1d5833d

  Copy full SHA for 1d5833d

• Return gradients as flattened list from custom-defined gradient op in jax2tf.

  Show description for 4d966bb
  

  ofirnachum

  authored and

  jax authors

  committed
  4d966bb

  Copy full SHA for 4d966bb

• Merge pull request jax-ml#9891 from gnecula:jax_poly_doc

  Show description for 1f0a5b3
  

  jax authors

  committed
  1f0a5b3

  Copy full SHA for 1f0a5b3

Pagination

Previous

Next