Commits

Branch selector

 87122fc

User selector

All users

Datepicker

All time

Commit History

Commits on Jan 29, 2021

• Update jax2tf.py

  Show description for 87122fc
  gnecula

  authored
  87122fc

  Copy full SHA for 87122fc

• Update jax2tf.py

  Show description for 3fa950f
  gnecula

  authored
  3fa950f

  Copy full SHA for 3fa950f

• Merge branch 'master' into changelist/330700645

  gnecula

  authored
  8c35db7

  Copy full SHA for 8c35db7

• Merge pull request jax-ml#5548 from jakevdp:fix-doc-sigs

  Show description for 580e107
  

  jax authors

  committed
  580e107

  Copy full SHA for 580e107

• Merge pull request jax-ml#5547 from skye:fix

  Show description for 8719609
  

  jax authors

  committed
  8719609

  Copy full SHA for 8719609

• Remove signatures from wrapped docstrings

  jakevdp

  committed
  ba9015a

  Copy full SHA for ba9015a

Commits on Jan 28, 2021

• Merge pull request jax-ml#5545 from jakevdp:fix-doc

  Show description for 3d5d073
  

  jax authors

  committed
  3d5d073

  Copy full SHA for 3d5d073

• Improve error message when a reduction function returns an invalid return type.

  Show description for 997e6ef
  

  skye

  and

  mattjj

  committed
  997e6ef

  Copy full SHA for 997e6ef

• DOC: fix some minor formatting issues

  jakevdp

  committed
  af6da22

  Copy full SHA for af6da22

• Merge pull request jax-ml#5527 from google:remove-soft-pmap

  Show description for 543dcb3
  

  jax authors

  committed
  543dcb3

  Copy full SHA for 543dcb3

• Merge pull request jax-ml#5525 from google:pdot-tests4

  Show description for 8a58ca6
  

  jax authors

  committed
  8a58ca6

  Copy full SHA for 8a58ca6

• implement soft_pmap in terms of xmap

  mattjj

  committed
  014f9a8

  Copy full SHA for 014f9a8

• Merge pull request jax-ml#5540 from google:ci-pytest-verbose

  Show description for 9380e96
  

  jax authors

  committed
  9380e96

  Copy full SHA for 9380e96

• Merge pull request jax-ml#5538 from jakevdp:fix-doc

  Show description for dbc4ad4
  

  jax authors

  committed
  dbc4ad4

  Copy full SHA for dbc4ad4

• run CI pytest in verbose mode

  froystig

  committed
  db2298f

  Copy full SHA for db2298f

• DOC: add jax.experimental.maps to index

  jakevdp

  committed
  cbeb80e

  Copy full SHA for cbeb80e

Commits on Jan 27, 2021

• [avals with names] Revise aval constructor call sites to use a new aval.update method

  Show description for f1918f0
  

  jekbradbury

  authored and

  jax authors

  committed
  f1918f0

  Copy full SHA for f1918f0

• add systematic pdot vjp tests

  mattjj

  committed
  537c3d5

  Copy full SHA for 537c3d5

• Merge pull request jax-ml#5534 from cheshire:patch-1

  Show description for 7d9d149
  

  jax authors

  committed
  7d9d149

  Copy full SHA for 7d9d149

• Update jax2tf_limitations to update the list of integral dtypes for tf.pow

  Show description for 6b75cc6
  cheshire

  authored
  6b75cc6

  Copy full SHA for 6b75cc6

• Merge pull request jax-ml#5505 from apaszke:xmap-docs

  Show description for af76145
  

  jax authors

  committed
  af76145

  Copy full SHA for af76145

• Merge pull request jax-ml#5521 from chr1sj0nes:changelist/345649004

  Show description for deb2afe
  

  jax authors

  committed
  deb2afe

  Copy full SHA for deb2afe

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

  Show description for 1094740
  

  jax authors

  committed
  1094740

  Copy full SHA for 1094740

• Merge pull request jax-ml#5526 from google:issue5522

  Show description for a9e5fab
  

  jax authors

  committed
  a9e5fab

  Copy full SHA for a9e5fab

• update version and changelog for pypi

  mattjj

  committed
  d2ae949

  Copy full SHA for d2ae949

• add axis_env argument to make_jaxpr

  Show description for ff66c55
  mattjj

  committed
  ff66c55

  Copy full SHA for ff66c55

Commits on Jan 26, 2021

• Merge pull request jax-ml#5523 from sharadmv:custom-gradient-tuple

  Show description for 1fd1faa
  

  jax authors

  committed
  1fd1faa

  Copy full SHA for 1fd1faa

• Allow jax.custom_gradient to return vjp with singleton return value

  sharadmv

  committed
  6061b09

  Copy full SHA for 6061b09

• Merge pull request jax-ml#5490 from google:xeinsum

  Show description for 814c4ad
  

  jax authors

  committed
  814c4ad

  Copy full SHA for 814c4ad

• Merge pull request jax-ml#5426 from jakevdp:thinking-in-jax

  Show description for e2140bf
  

  jax authors

  committed
  e2140bf

  Copy full SHA for e2140bf

• Merge pull request jax-ml#5512 from google:closure-convert-docs

  Show description for eec2dbc
  

  jax authors

  committed
  eec2dbc

  Copy full SHA for eec2dbc

• relax test tolerance on tpu

  mattjj

  committed
  737c162

  Copy full SHA for 737c162

• Add thinking_in_jax.ipynb

  jakevdp

  committed
  967f3ac

  Copy full SHA for 967f3ac

• Add a docstring for xmap

  Show description for 83c8010
  apaszke

  committed
  83c8010

  Copy full SHA for 83c8010

• import closure_convert at top module level

  froystig

  committed
  93c61e4

  Copy full SHA for 93c61e4

Pagination

Previous

Next