Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Jun 27, 2023

• Merge pull request jax-ml#16566 from jakevdp:stats-binom

  Show description for 94c3e45
  

  jax authors

  committed
  94c3e45

• Add jax.scipy.stats.binom

  jakevdp

  committed
  30d1a8a

• Merge pull request jax-ml#16562 from froystig:outline-split

  Show description for 6bc74d2
  

  jax authors

  committed
  6bc74d2

• [shape_poly] Shape polymorphism support for approx_top_k

  Show description for cb42fae
  

  gnecula

  authored and

  jax authors

  committed
  cb42fae

• Make sharding on ShapeDtypeStruct a property that always exists. The previous behavior was it only existed if sharding was not None.

  Show description for 744a64f
  

  yashk2810

  authored and

  jax authors

  committed
  744a64f

• jax.lax.collapse now takes Nones for stop_dimension.

  Show description for 819f731
  

  pschuh

  authored and

  jax authors

  committed
  819f731

• Raise an error if a user passes None to host_local_array_to_global_array or global_array_to_host_local_array

  Show description for c632cac
  

  yashk2810

  authored and

  jax authors

  committed
  c632cac

Commits on Jun 26, 2023

• [shape_poly] linalg.schur: shape polymorphism with native lowering on CPU

  Show description for c6a6005
  

  gnecula

  authored and

  jax authors

  committed
  c6a6005

• [shape_poly] linalg.triangular_solve: shape polymorphism with native lowering on CPU

  Show description for a91412e
  

  gnecula

  authored and

  jax authors

  committed
  a91412e

• outline jitted threefry split and fold_in subroutines

  Show description for 690e626
  froystig

  committed
  690e626

• [shape_poly] Refactor support for dynamic shapes for linalg.eig and linalg.eigh

  Show description for ea0e50f
  

  gnecula

  authored and

  jax authors

  committed
  ea0e50f

• Merge pull request jax-ml#16554 from gnecula:poly_refactor

  Show description for be0f4ce
  

  jax authors

  committed
  be0f4ce

• Merge pull request jax-ml#16467 from gnecula:safety_checks

  Show description for dad18c1
  

  jax authors

  committed
  dad18c1

• [shape_poly] Refactor some older support for shape polymorphism for linalg.

  Show description for aadcec2
  

  gnecula

  authored and

  jax authors

  committed
  aadcec2

• [PJRT C API] Add related C type definitions for key value get/put callback, as well as conversion between C and cpp types.

  Show description for 21588a3
  

  Jieying Luo

  authored and

  jax authors

  committed
  21588a3

Commits on Jun 25, 2023

• [jax2tf] Improve custom call safety check to detect mhlo.custom_call

  gnecula

  committed
  c800ef8

• [shape_poly] Cleanup the evaluation of dynamic shapes

  Show description for 2299f05
  gnecula

  committed
  2299f05

Commits on Jun 24, 2023

• Make the default driver in serialization be a global constant.

  Show description for feced36
  

  pschuh

  authored and

  jax authors

  committed
  feced36

Commits on Jun 23, 2023

• Use lower-case PEP 585 names for types.

  Show description for 816ba91
  

  hawkinsp

  authored and

  jax authors

  committed
  816ba91

• Merge pull request jax-ml#16430 from jakevdp:bool-error

  Show description for f67acee
  

  jax authors

  committed
  f67acee

• Merge pull request jax-ml#16533 from hawkinsp:winreadme

  Show description for 9d480e2
  

  jax authors

  committed
  9d480e2

• Update README to mention Windows support.

  hawkinsp

  committed
  a1de687

• Merge pull request jax-ml#16487 from jakevdp:convolve-dtype

  Show description for 01a16f5
  

  jax authors

  committed
  01a16f5

• Merge pull request jax-ml#16535 from jakevdp:numpy-122

  Show description for 1e1992a
  

  jax authors

  committed
  1e1992a

• Drop support for numpy 1.21

  Show description for ad35702
  jakevdp

  committed
  ad35702

• Merge pull request jax-ml#16386 from axch:ragged-einsum

  Show description for 63415a9
  

  jax authors

  committed
  63415a9

• Relax test tolerances.

  Show description for 0adfafe
  

  hawkinsp

  authored and

  jax authors

  committed
  0adfafe

• Remove references to Python 3.8.

  Show description for bfa113b
  

  hawkinsp

  authored and

  jax authors

  committed
  bfa113b

• Type annotations.

  axch

  committed
  bb7d918

• [shape_poly] linalg.lu: for shape polymorphism for native serialization on CPU.

  Show description for bbc6f30
  

  gnecula

  authored and

  jax authors

  committed
  bbc6f30

• Merge pull request jax-ml#16514 from ayaka14732:main

  Show description for 935579d
  

  jax authors

  committed
  935579d

• Merge pull request jax-ml#16503 from jakevdp:sparsify-custom-jvp

  Show description for 7205e55
  

  jax authors

  committed
  7205e55

• Merge pull request jax-ml#16428 from jakevdp:prng-tests

  Show description for 1b66d97
  

  jax authors

  committed
  1b66d97

• Custom PRNG: improve test coverage when enable_custom_prng=false

  Show description for 39645b5
  jakevdp

  committed
  39645b5

• [sparse] support custom JVP in sparsify

  jakevdp

  committed
  b6d5445

Pagination

Previous

Next