Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Jun 15, 2022

• Relax typechecking for preferred_element_type, to allow integer->floating dot products.

  Show description for b51ee37
  

  reinerp

  authored and

  jax authors

  committed
  b51ee37

  Copy full SHA for b51ee37

• Give auto sharding the logical device ids instead of the physical device ids

  Show description for f195f1e
  

  yashk2810

  authored and

  jax authors

  committed
  f195f1e

  Copy full SHA for f195f1e

• Merge pull request jax-ml#11043 from nicholasjng:xor-reduce

  Show description for 07c6abb
  

  jax authors

  committed
  07c6abb

  Copy full SHA for 07c6abb

• Merge pull request jax-ml#11101 from jakevdp:x64-tree-util-test

  Show description for 705b2ae
  

  jax authors

  committed
  705b2ae

  Copy full SHA for 705b2ae

• Merge pull request jax-ml#11107 from hawkinsp:xla

  Show description for 8aeeba0
  

  jax authors

  committed
  8aeeba0

  Copy full SHA for 8aeeba0

• Add bitwise XOR reducer to lax.reduce

  Show description for 311e6a9
  nicholasjng

  committed
  311e6a9

  Copy full SHA for 311e6a9

• Update XLA pin.

  hawkinsp

  committed
  5b7c485

  Copy full SHA for 5b7c485

• Merge pull request jax-ml#11105 from hawkinsp:workspacecomment

  Show description for acc7dc0
  

  jax authors

  committed
  acc7dc0

  Copy full SHA for acc7dc0

• Improve the comment in WORKSPACE describing how to override the TF repository.

  hawkinsp

  committed
  1e504b3

  Copy full SHA for 1e504b3

Commits on Jun 14, 2022

• Add jax.config.jax_default_device to jax in-memory cache key

  Show description for 7098088
  

  skye

  authored and

  jax authors

  committed
  7098088

  Copy full SHA for 7098088

• Merge pull request jax-ml#11080 from google:dependabot/github_actions/actions/setup-python-4

  Show description for 3b5bc51
  

  jax authors

  committed
  3b5bc51

  Copy full SHA for 3b5bc51

• [mhlo] ConstOp -> ConstantOp, a la HLO

  Show description for aeabfc4
  

  atondwal

  authored and

  jax authors

  committed
  aeabfc4

  Copy full SHA for aeabfc4

• Merge pull request jax-ml#11032 from JiahaoYao:patch-1

  Show description for 4244c4f
  

  jax authors

  committed
  4244c4f

  Copy full SHA for 4244c4f

• Merge pull request jax-ml#11001 from SiegeLordEx:zero_reductions

  Show description for fdcfb30
  

  jax authors

  committed
  fdcfb30

  Copy full SHA for fdcfb30

• Merge pull request jax-ml#11098 from sshahrokhi:telemetry

  Show description for f0cea24
  

  jax authors

  committed
  f0cea24

  Copy full SHA for f0cea24

• [x64] make tree_util_test compatible with strict dtype promotion

  jakevdp

  committed
  6efb03c

  Copy full SHA for 6efb03c

• Allow doing reductions on empty arrays in some cases.

  Show description for ff637e1
  SiegeLordEx

  committed
  ff637e1

  Copy full SHA for ff637e1

• adding os env to track JAX platform

  sshahrokhi

  committed
  88f1b9f

  Copy full SHA for 88f1b9f

• Merge pull request jax-ml#10424 from NeilGirdhar:vmap_annotation

  Show description for cd565f8
  

  jax authors

  committed
  cd565f8

  Copy full SHA for cd565f8

• Merge pull request jax-ml#10865 from jakevdp:x64-lax-numpy-test

  Show description for cf12937
  

  jax authors

  committed
  cf12937

  Copy full SHA for cf12937

• Merge pull request jax-ml#11052 from jakevdp:x64-lax-scipy-test

  Show description for d418e89
  

  jax authors

  committed
  d418e89

  Copy full SHA for d418e89

• Annotate vmap

  NeilGirdhar

  committed
  7697616

  Copy full SHA for 7697616

• Fix the _check_if_deleted check that was merged at the wrong place by the cider merging machinery.

  Show description for b3130b7
  

  yashk2810

  authored and

  jax authors

  committed
  b3130b7

  Copy full SHA for b3130b7

• Merge pull request jax-ml#11051 from carlosgmartin:gennorm

  Show description for 99a5817
  

  jax authors

  committed
  99a5817

  Copy full SHA for 99a5817

• [x64] fix lax_numpy_test for strict dtypes

  jakevdp

  committed
  9401247

  Copy full SHA for 9401247

• Added scipy.stats.gennorm.

  carlosgmartin

  committed
  57b89ba

  Copy full SHA for 57b89ba

• Make Shard a normal class making index and replica_id properties on that class. Raise an error if an indices cannot be calculated from a sharding.

  Show description for 5b808c9
  

  yashk2810

  authored and

  jax authors

  committed
  5b808c9

  Copy full SHA for 5b808c9

• Merge pull request jax-ml#11046 from jakevdp:x64-weak-type-inexact

  Show description for 9eb27a4
  

  jax authors

  committed
  9eb27a4

  Copy full SHA for 9eb27a4

• [x64] make lax_scipy_test.py compatible with strict dtype promotion

  jakevdp

  committed
  e888e7c

  Copy full SHA for e888e7c

• [x64] preserve weak types in promote_dtypes_inexact

  jakevdp

  committed
  e8690f6

  Copy full SHA for e8690f6

• Merge pull request jax-ml#11064 from hawkinsp:cacheclear

  Show description for 691a1f8
  

  jax authors

  committed
  691a1f8

  Copy full SHA for 691a1f8

• Merge pull request jax-ml#11085 from hawkinsp:remat

  Show description for 8ab5037
  

  jax authors

  committed
  8ab5037

  Copy full SHA for 8ab5037

• Merge pull request jax-ml#11073 from ludgerpaehler:main

  Show description for 086a88a
  

  jax authors

  committed
  086a88a

  Copy full SHA for 086a88a

• Add __array__ (for device_get), _npy_value, block_until_ready, delete and _check_if_deleted to Array.

  Show description for a716065
  

  yashk2810

  authored and

  jax authors

  committed
  a716065

  Copy full SHA for a716065

• Change the remat rule registration to register a default with a GPU specialization.

  Show description for aae525c
  hawkinsp

  committed
  aae525c

  Copy full SHA for aae525c

Pagination

Previous

Next