Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Nov 7, 2022

• Merge pull request jax-ml#13093 from PhilipVinc:patch-1

  Show description for e00f7e7
  

  jax authors

  committed
  e00f7e7

• Add UnloadedMeshExecutable to represent a MeshExecutable that is not loaded

  Show description for 2c1fe45
  

  pschuh

  authored and

  jax authors

  committed
  2c1fe45

• Fix issue in check_tree, so that custom_linear_solve supports

  Show description for 793fb9b
  PhilipVinc

  committed
  793fb9b

• Fix GDA error message formatting

  Show description for 2183059
  

  hyeontaek

  authored and

  jax authors

  committed
  2183059

• Remove static_argnums from AOT invocation.

  Show description for b127b70
  

  cky9301

  authored and

  jax authors

  committed
  b127b70

• Check for ArrayImpl rather than sharding because this code is supposed to check for concrete Array until a shard_like primitive exists.

  Show description for da519f3
  

  yashk2810

  authored and

  jax authors

  committed
  da519f3

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

  Show description for 587885b
  

  jax authors

  committed
  587885b

• Merge pull request jax-ml#13146 from jakevdp:fix-flake8

  Show description for 3da554d
  

  jax authors

  committed
  3da554d

• Remove whitespace to fix flake8

  jakevdp

  committed
  b0e03fb

• Add a number of missing function cross-references in the docs.

  hawkinsp

  committed
  cd84eb1

• Merge pull request jax-ml#12890 from ROCmSoftwarePlatform:rocm_enable_multi_gpu_test

  Show description for 042595b
  

  jax authors

  committed
  042595b

• Merge pull request jax-ml#13140 from gnecula:clean_limitations

  Show description for e9e014f
  

  jax authors

  committed
  e9e014f

• [jax2tf] Fixed jax2tf Limitations

  Show description for d9b1dc3
  gnecula

  committed
  d9b1dc3

• Merge pull request jax-ml#13108 from mattjj:djax-vmap2

  Show description for 8d59b0d
  

  jax authors

  committed
  8d59b0d

• add a basic prototype of piles, behind jax_dynamic_shapes

  Show description for f2f2faa
  

  3 peoplecommitted
  f2f2faa

Commits on Nov 6, 2022

• If the input to host_local_array_to_global_array is not fully addressable (i.e. not host local), return it as is.

  Show description for 2a262e9
  

  yashk2810

  authored and

  jax authors

  committed
  2a262e9

• Remove device_indices method which is redundant because of the existence of devices_indices_map and is slower because pinging a cache for every device is not free.

  Show description for 3dbe101
  

  yashk2810

  authored and

  jax authors

  committed
  3dbe101

Commits on Nov 5, 2022

• Make Sharding instances picklable.

  Show description for 5b45357
  

  tomhennigan

  authored and

  jax authors

  committed
  5b45357

• Improve the sharding mismatch error message by adding the arg to the message too.

  Show description for d9f3dc3
  

  yashk2810

  authored and

  jax authors

  committed
  d9f3dc3

• Merge pull request jax-ml#13127 from mattjj:issue13124

  Show description for 0c36f84
  

  jax authors

  committed
  0c36f84

• Improve the error message when the avals a function was AOT compiled with doesn't match the input avals when its called.

  Show description for e161d20
  

  yashk2810

  authored and

  jax authors

  committed
  e161d20

• fix jax.random.logits shape argument

  Show description for 190204f
  mattjj

  committed
  190204f

Commits on Nov 4, 2022

• Merge pull request jax-ml#13122 from yejingxin:main

  Show description for 2932c1e
  

  jax authors

  committed
  2932c1e

• Merge pull request jax-ml#13024 from treyra:patch-2

  Show description for 75b5ccc
  

  jax authors

  committed
  75b5ccc

• update pytest.ini to print warning message for compilation_cache_test

  yejingxin

  committed
  e6c88f2

• Merge pull request jax-ml#13113 from jakevdp:mypy-fix

  Show description for 2ce7eb5
  

  jax authors

  committed
  2ce7eb5

• Add a section to mention the unsafe_rbg approach for reducing the size of the generated TF graph.

  Show description for f056647
  

  voutcn

  authored and

  jax authors

  committed
  f056647

• Add x86_64 dependency note to pip installation

  Show description for e3cd967
  treyra

  committed
  e3cd967

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

  Show description for 1799cf9
  

  jax authors

  committed
  1799cf9

• CI: add absl-py to docs/requirements.txt

  jakevdp

  committed
  acdb545

• [typing] update jaxlib & remove unnecessary ignore

  jakevdp

  committed
  0691be6

• Merge pull request jax-ml#13114 from skye:runner_readme

  Show description for 2f1f3e4
  

  jax authors

  committed
  2f1f3e4

• Add .github/workflows/self_hosted_runner_utils/README.md

  Show description for 52775c4
  skye

  committed
  52775c4

• Finish the release of jax and jaxlib 0.3.24

  Show description for 1d48c93
  

  yashk2810

  authored and

  jax authors

  committed
  1d48c93

• Merge pull request jax-ml#13097 from jakevdp:actions-permissions

  Show description for 3db2a59
  

  jax authors

  committed
  3db2a59

Pagination

Previous

Next