Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Apr 5, 2022

• register a backend for plaidml xla cpu

  ronghongbo

  committed
  33b254f

  Copy full SHA for 33b254f

Commits on Apr 2, 2022

• Merge pull request jax-ml#10122 from sharadmv:jax2tf-name-stack

  Show description for 359b614
  

  jax authors

  committed
  359b614

  Copy full SHA for 359b614

• Merge pull request jax-ml#10121 from hawkinsp:hcbcache

  Show description for e64a57d
  

  jax authors

  committed
  e64a57d

  Copy full SHA for e64a57d

• Merge pull request jax-ml#10126 from jakevdp:tree-multimap

  Show description for cdf4177
  

  jax authors

  committed
  cdf4177

  Copy full SHA for cdf4177

Commits on Apr 1, 2022

• DOC: switch from tree_multimap to tree_map in docs

  jakevdp

  committed
  c61a18b

  Copy full SHA for c61a18b

• Deprecate jax.tree_util.tree_multimap

  jakevdp

  committed
  df1ceae

  Copy full SHA for df1ceae

• Merge pull request jax-ml#10123 from fabianp:patch-4

  Show description for 9693898
  

  jax authors

  committed
  9693898

  Copy full SHA for 9693898

• Give auto sharder the mesh information specifically the mesh_shape and the devices

  Show description for 0eaeff6
  

  yashk2810

  authored and

  jax authors

  committed
  0eaeff6

  Copy full SHA for 0eaeff6

• remove $ from command line commands

  Show description for 4fd466b
  fabianp

  authored
  4fd466b

  Copy full SHA for 4fd466b

• Merge pull request jax-ml#10110 from pschuh:weakref-bug

  Show description for 1c3edc8
  

  jax authors

  committed
  1c3edc8

  Copy full SHA for 1c3edc8

• Fixes jax2tf's test_name_scope to use graph introspection instead of

  Show description for aac8ec8
  sharadmv

  committed
  aac8ec8

  Copy full SHA for aac8ec8

• Give auto sharder the mesh information specifically the mesh_shape and the devices

  Show description for 7b7458b
  

  yashk2810

  authored and

  jax authors

  committed
  7b7458b

  Copy full SHA for 7b7458b

• Fix race condition for weakref destructor by catching rare exceptions.

  pschuh

  committed
  df1c478

  Copy full SHA for df1c478

• Expose AutoSharding's mesh_shape and mesh_ids options to JAX.

  Show description for 8c3385c
  

  jax authors

  committed
  8c3385c

  Copy full SHA for 8c3385c

• Avoid retracing when a host_callback.call is called multiple times with the same function.

  Show description for 208e83c
  hawkinsp

  committed
  208e83c

  Copy full SHA for 208e83c

• Merge pull request jax-ml#10119 from jakevdp:pil-fix

  Show description for a4a551a
  

  jax authors

  committed
  a4a551a

  Copy full SHA for a4a551a

• CI: pin pillow<9.1 to prevent deprecation warnings

  jakevdp

  committed
  1f300e7

  Copy full SHA for 1f300e7

• Merge pull request jax-ml#10058 from yotarok:istft

  Show description for e766b96
  

  jax authors

  committed
  e766b96

  Copy full SHA for e766b96

• Merge pull request jax-ml#10103 from LenaMartens:changelist/438319917

  Show description for 4decbcb
  

  jax authors

  committed
  4decbcb

  Copy full SHA for 4decbcb

• Fix the breakage by including --experimental_cc_shared_library as done by TF.

  Show description for aa5d6b4
  

  yashk2810

  authored and

  jax authors

  committed
  aa5d6b4

  Copy full SHA for aa5d6b4

• Add istft to jax.scipy.signal.

  yotarok

  committed
  a7fd751

  Copy full SHA for a7fd751

• Merge pull request jax-ml#10113 from mihaimaruseac:patch-1

  Show description for d42981b
  

  jax authors

  committed
  d42981b

  Copy full SHA for d42981b

• First attempt to enable auto-sharding. This CL adds support for GDA (no SDA support yet).

  Show description for 8ca8f74
  

  yashk2810

  authored and

  jax authors

  committed
  8ca8f74

  Copy full SHA for 8ca8f74

• Add missing :

  mihaimaruseac

  authored
  9b68e1a

  Copy full SHA for 9b68e1a

• Fix syntax error causing typo

  mihaimaruseac

  authored
  113cfe7

  Copy full SHA for 113cfe7

• Manually split the version string

  Show description for f09aff1
  mihaimaruseac

  authored
  f09aff1

  Copy full SHA for f09aff1

Commits on Mar 31, 2022

• Don't compare versions as strings.

  Show description for cdd703b
  mihaimaruseac

  authored
  cdd703b

  Copy full SHA for cdd703b

• [sparse] Update docstrings for bcoo primitives.

  Show description for 3184dd6
  

  tlu7

  authored and

  jax authors

  committed
  3184dd6

  Copy full SHA for 3184dd6

• Merge pull request jax-ml#10097 from lgeiger:expand-dims

  Show description for ced2cbe
  

  jax authors

  committed
  ced2cbe

  Copy full SHA for ced2cbe

• Merge pull request jax-ml#10092 from jakevdp:sharp-bits-divergences

  Show description for d9403f6
  

  jax authors

  committed
  d9403f6

  Copy full SHA for d9403f6

• Merge pull request jax-ml#10102 from gnecula:hcb_fix

  Show description for 5181692
  

  jax authors

  committed
  5181692

  Copy full SHA for 5181692

• Sharp bits: add miscellaneous divergences from numpy

  jakevdp

  committed
  72f14a5

  Copy full SHA for 72f14a5

• [jax2tf] Add links to Google-internal documentation.

  Show description for f31b7c9
  

  gnecula

  authored and

  jax authors

  committed
  f31b7c9

  Copy full SHA for f31b7c9

• Merge pull request jax-ml#10089 from sharadmv:name-stack-fix

  Show description for 137df31
  

  jax authors

  committed
  137df31

  Copy full SHA for 137df31

• If jax_parallel_functions_output_gda flag is set to True, then all outputs are GDA. In abstract_eval rule of pjit, don't convert global avals to local avals if the `jax_parallel_functions_outpu…

  Show description for 0bcd447
  

  yashk2810

  authored and

  jax authors

  committed
  0bcd447

  Copy full SHA for 0bcd447

Pagination

Previous

Next