Commits

Breadcrumbs

History for

on6715736

User selector

All users

Datepicker

All time

Commit History

Commits on Mar 25, 2023

• Add src argument to device_put as an experimental arg

  Show description for a5d3085
  

  yashk2810

  authored and

  jax authors

  committed
  a5d3085

  Copy full SHA for a5d3085

• Merge pull request jax-ml#15212 from google:pjrt_c_api_tests

  Show description for c572155
  

  jax authors

  committed
  c572155

  Copy full SHA for c572155

Commits on Mar 24, 2023

• Remove 'pjrt_c_api_unimplemented' pytest mark.

  Show description for ef5e4a4
  skye

  committed
  ef5e4a4

  Copy full SHA for ef5e4a4

• Migrate regularized_incomplete_beta_p off xla_fallback

  Show description for 6842e98
  

  atondwal

  authored and

  jax authors

  committed
  6842e98

  Copy full SHA for 6842e98

• Migrate besseli0e off xla_fallback

  Show description for ac44d2c
  

  atondwal

  authored and

  jax authors

  committed
  ac44d2c

  Copy full SHA for ac44d2c

• Delete remote TPU support.

  Show description for 6ed66ad
  

  hawkinsp

  authored and

  jax authors

  committed
  6ed66ad

  Copy full SHA for 6ed66ad

• Guard ArrayImpl checks by xla_extension_version.

  Show description for 21541e6
  

  pschuh

  authored and

  jax authors

  committed
  21541e6

  Copy full SHA for 21541e6

• After the SPMD bug fix, always take the _rewriting_take route for getitem instead of bouncing to host.

  Show description for bc231ee
  

  yashk2810

  authored and

  jax authors

  committed
  bc231ee

  Copy full SHA for bc231ee

• Migrate random_gamma_grad off xla_fallback

  Show description for 8c75e27
  

  atondwal

  authored and

  jax authors

  committed
  8c75e27

  Copy full SHA for 8c75e27

• Migrate igamma_grad_a_p off xla_fallback

  Show description for 8d1d522
  

  atondwal

  authored and

  jax authors

  committed
  8d1d522

  Copy full SHA for 8d1d522

• Migrate igammac_p off xla_fallback path

  Show description for 4a9b094
  

  atondwal

  authored and

  jax authors

  committed
  4a9b094

  Copy full SHA for 4a9b094

• Merge pull request jax-ml#15192 from mattjj:issue15190

  Show description for 32e7128
  

  jax authors

  committed
  32e7128

  Copy full SHA for 32e7128

• Merge pull request jax-ml#15187 from mattjj:djax-revival

  Show description for 1982a11
  

  jax authors

  committed
  1982a11

  Copy full SHA for 1982a11

• [dynamic-shapes] make dynamic shape staging-to-jaxpr work with pjit

  mattjj

  committed
  7743fcd

  Copy full SHA for 7743fcd

• fix jax.Array.round()

  Show description for 793387e
  mattjj

  committed
  793387e

  Copy full SHA for 793387e

• Remove PJRT C API bypass.

  Show description for 4cb3b01
  

  skye

  authored and

  jax authors

  committed
  4cb3b01

  Copy full SHA for 4cb3b01

• Increase minimum NumPy version to 1.21.

  Show description for b7375b3
  hawkinsp

  committed
  b7375b3

  Copy full SHA for b7375b3

• Merge pull request jax-ml#15184 from jakevdp:move-median

  Show description for e9bc7ee
  

  jax authors

  committed
  e9bc7ee

  Copy full SHA for e9bc7ee

Commits on Mar 23, 2023

• lax_numpy: move quantile-based functions to reductions.py

  jakevdp

  committed
  6f8885a

  Copy full SHA for 6f8885a

• Migrate igamma_p off xla_fallback

  Show description for f981243
  

  atondwal

  authored and

  jax authors

  committed
  f981243

  Copy full SHA for f981243

• Add padding option "SAME_LOWER" for ticket jax-ml#14990

  Show description for 171b22d
  

  maxwillzq

  authored and

  jax authors

  committed
  171b22d

  Copy full SHA for 171b22d

• improve scan error messages

  mattjj

  committed
  ba2ff51

  Copy full SHA for ba2ff51

• Deprecated xla_call_p since it has been replaced with pjit.pjit_p

  Show description for a9e48af
  

  yashk2810

  authored and

  jax authors

  committed
  a9e48af

  Copy full SHA for a9e48af

• Merge pull request jax-ml#14937 from b0nce:fix-stats

  Show description for 383cf41
  

  jax authors

  committed
  383cf41

  Copy full SHA for 383cf41

• Merge pull request jax-ml#15123 from jakevdp:fix-mean-large-dims

  Show description for 54e8101
  

  jax authors

  committed
  54e8101

  Copy full SHA for 54e8101

• Merge pull request jax-ml#15154 from mattjj:pjit-typecheck

  Show description for e39578c
  

  jax authors

  committed
  e39578c

  Copy full SHA for e39578c

• Redefine compile_and_serialize as serialize(lowered.compile()).

  Show description for 484eb26
  

  pschuh

  authored and

  jax authors

  committed
  484eb26

  Copy full SHA for 484eb26

Commits on Mar 22, 2023

• enable pjit recursive typechecking

  Show description for 268456e
  mattjj

  committed
  268456e

  Copy full SHA for 268456e

• Merge pull request jax-ml#15151 from mattjj:qiao-remat-print-res-info

  Show description for 5f724cf
  

  jax authors

  committed
  5f724cf

  Copy full SHA for 5f724cf

• add experimental jax_log_checkpoint_residuals option

  Show description for 6b4262d
  

  mattjj

  and

  zhangqiaorjc

  committed
  6b4262d

  Copy full SHA for 6b4262d

• Add print statement to help debug spurious test failure

  sharadmv

  committed
  12761f9

  Copy full SHA for 12761f9

• Merge pull request jax-ml#15085 from mattjj:arg-info-in-mlir-5

  Show description for dd2ecf4
  

  jax authors

  committed
  dd2ecf4

  Copy full SHA for dd2ecf4

• jax.random: remove scale from wald function

  jakevdp

  committed
  398532b

  Copy full SHA for 398532b

• Revert: custom_vjp symbolic zeros support

  Show description for 64e1f5f
  

  hawkinsp

  authored and

  jax authors

  committed
  64e1f5f

  Copy full SHA for 64e1f5f

• Merge pull request jax-ml#15114 from JiaYaobo:add_wald_random

  Show description for 00e6c73
  

  jax authors

  committed
  00e6c73

  Copy full SHA for 00e6c73

Pagination

Previous

Next