Commits

Branch selector

 2f105bd

User selector

All users

Datepicker

All time

Commit History

Commits on Mar 28, 2023

• Jax 0.4.7 has been released so assert that length of warnings is 1 in test_cache_read_warning

  Show description for 2f105bd
  

  yashk2810

  authored and

  jax authors

  committed
  2f105bd

  Copy full SHA for 2f105bd

• Merge pull request jax-ml#15251 from jakevdp:mypy-deps

  Show description for 2fbccc8
  

  jax authors

  committed
  2fbccc8

  Copy full SHA for 2fbccc8

• Remove MeshPspecSharding since it has been more than 3 months since it was deprecated (Nov 2, 2022). The replacement name is NamedSharding.

  Show description for 7442faa
  

  yashk2810

  authored and

  jax authors

  committed
  7442faa

  Copy full SHA for 7442faa

• Deprecate FROM_GDA and remove its support from pjit's code since jax.Array inside pjit has sharding inference capabilities by default.

  Show description for 97c8ce3
  

  yashk2810

  authored and

  jax authors

  committed
  97c8ce3

  Copy full SHA for 97c8ce3

• Set coordinator address to allow it to later be used to initialize OCDBT coordinator server. Allow user to pass ts.Context when serializing or deserializing.

  Show description for 6ace666
  

  cpgaffney1

  authored and

  jax authors

  committed
  6ace666

  Copy full SHA for 6ace666

• Remove jax2tf experimental_native_lowering.

  Show description for 2ac2dc6
  

  gnecula

  authored and

  jax authors

  committed
  2ac2dc6

  Copy full SHA for 2ac2dc6

• Remove Cuda 11.4 support. JAX from 0.4.8 release will support cuda 11.8 and cuda 12

  Show description for 86c0b36
  

  yashk2810

  authored and

  jax authors

  committed
  86c0b36

  Copy full SHA for 86c0b36

• Merge pull request jax-ml#15258 from gnecula:dim_vars

  Show description for 4106c35
  

  jax authors

  committed
  4106c35

  Copy full SHA for 4106c35

• [shape_poly] Refactor the computation of the dimension variables in native serialization

  Show description for a1538c7
  gnecula

  committed
  a1538c7

  Copy full SHA for a1538c7

• Merge pull request jax-ml#15206 from jakevdp:expm-batch

  Show description for 4533578
  

  jax authors

  committed
  4533578

  Copy full SHA for 4533578

• Merge pull request jax-ml#15241 from jakevdp:instance-check

  Show description for 03d1442
  

  jax authors

  committed
  03d1442

  Copy full SHA for 03d1442

Commits on Mar 27, 2023

• Merge pull request jax-ml#15243 from jakevdp:coo-sort-warning

  Show description for 6f39237
  

  jax authors

  committed
  6f39237

  Copy full SHA for 6f39237

• jax.scipy.linalg.expm: support batched inputs

  jakevdp

  committed
  ad0fc89

  Copy full SHA for ad0fc89

• CI: add numpy & scipy to mypy env

  jakevdp

  committed
  61190cb

  Copy full SHA for 61190cb

• Finish jax and jaxlib 0.4.7 release

  Show description for 670fba3
  

  yashk2810

  authored and

  jax authors

  committed
  670fba3

  Copy full SHA for 670fba3

• Add jaxlib version guard for rnn test

  Show description for 10dc941
  

  sharadmv

  authored and

  jax authors

  committed
  10dc941

  Copy full SHA for 10dc941

• Move jax.interpreters.partial_eval to jax._src.interpreters.partial_eval.

  Show description for 6cc1bf5
  

  hawkinsp

  authored and

  jax authors

  committed
  6cc1bf5

  Copy full SHA for 6cc1bf5

• Update the commit in workspace too

  Show description for ae4f1fc
  

  yashk2810

  authored and

  jax authors

  committed
  ae4f1fc

  Copy full SHA for ae4f1fc

• Prepare for jax and jaxlib 0.4.7 release

  Show description for e9cac5e
  

  yashk2810

  authored and

  jax authors

  committed
  e9cac5e

  Copy full SHA for e9cac5e

• Add deprecation warning for FROM_GDA usage since that argument is not required anymore.

  Show description for e21aee1
  

  yashk2810

  authored and

  jax authors

  committed
  e21aee1

  Copy full SHA for e21aee1

• Copy seq_lengths before creating descriptor

  Show description for 3c3fa04
  

  sharadmv

  authored and

  jax authors

  committed
  3c3fa04

  Copy full SHA for 3c3fa04

• Use pytype_strict_library() in Bazel build rules.

  Show description for 88c2898
  

  hawkinsp

  authored and

  jax authors

  committed
  88c2898

  Copy full SHA for 88c2898

• [sparse] fix coo efficiency warning

  jakevdp

  committed
  392bd93

  Copy full SHA for 392bd93

• jax.typing: recommend instance check in Python 3.10 or newer

  jakevdp

  committed
  ed9fa13

  Copy full SHA for ed9fa13

• Fix duplicate definition of 'cuda' extra in setup.py.

  Show description for 40fb646
  

  hawkinsp

  authored and

  jax authors

  committed
  40fb646

  Copy full SHA for 40fb646

• Merge pull request jax-ml#8633 from shawwn:2021-11-19/autodidax-fix-jaxpr-subcomp-return-type

  Show description for af4d494
  

  jax authors

  committed
  af4d494

  Copy full SHA for af4d494

• Merge pull request jax-ml#15218 from hawkinsp:mypy

  Show description for 10d51c7
  

  jax authors

  committed
  10d51c7

  Copy full SHA for 10d51c7

• Temporarily fix the compilation cache test which is failing on latest jaxlib release

  Show description for 41695cc
  

  yashk2810

  authored and

  jax authors

  committed
  41695cc

  Copy full SHA for 41695cc

• Merge pull request jax-ml#15226 from canyon289:patch-1

  Show description for 2c4be6f
  

  jax authors

  committed
  2c4be6f

  Copy full SHA for 2c4be6f

• Merge pull request jax-ml#15228 from canyon289:patch-2

  Show description for d19e60e
  

  jax authors

  committed
  d19e60e

  Copy full SHA for d19e60e

• Delete benchmark and pmap_benchmark files as they are legacy and replaced with api_benchmark.py

  Show description for cf8c2b8
  

  yashk2810

  authored and

  jax authors

  committed
  cf8c2b8

  Copy full SHA for cf8c2b8

• Merge pull request jax-ml#13008 from hawkinsp:pipcuda

  Show description for d473e86
  

  jax authors

  committed
  d473e86

  Copy full SHA for d473e86

• Merge pull request jax-ml#15205 from yhtang:editable-jaxlib-build

  Show description for 6715736
  

  jax authors

  committed
  6715736

  Copy full SHA for 6715736

• Merge pull request jax-ml#15234 from gnecula:get_dim_size

  Show description for f3613a1
  

  jax authors

  committed
  f3613a1

  Copy full SHA for f3613a1

• [shape_poly] Fixed bug with dimension variables in unused args

  Show description for befb449
  gnecula

  committed
  befb449

  Copy full SHA for befb449

Pagination

Previous

Next