Commits

Branch selector

 8f93629

User selector

All users

Datepicker

All time

Commit History

Commits on Mar 10, 2022

• remove _convert_element_type from public jax.lax module

  froystig

  committed
  8f93629

  Copy full SHA for 8f93629

• Support gamma distribution with PRNGKeys other than threefry2x32.

  Show description for caf094d
  

  jpuigcerver

  authored and

  jax authors

  committed
  caf094d

  Copy full SHA for caf094d

Commits on Mar 9, 2022

• Add the input avals to Lowered and Compiled.

  Show description for 8a85544
  

  jblespiau

  authored and

  jax authors

  committed
  8a85544

  Copy full SHA for 8a85544

• Merge pull request jax-ml#9814 from hawkinsp:nnjit

  Show description for 41a8de6
  

  jax authors

  committed
  41a8de6

  Copy full SHA for 41a8de6

• Add @jit decorators to functions in jax.nn.

  hawkinsp

  committed
  626c03f

  Copy full SHA for 626c03f

• Merge pull request jax-ml#9807 from jakevdp:coo-sorted-cols

  Show description for beb9900
  

  jax authors

  committed
  beb9900

  Copy full SHA for beb9900

• Add JAX support for pjit on CPU

  Show description for 6e63a90
  

  yashk2810

  authored and

  jax authors

  committed
  6e63a90

  Copy full SHA for 6e63a90

• [sparse] track sorted columns for COO GPU lowerings

  jakevdp

  committed
  3679e0c

  Copy full SHA for 3679e0c

• Merge pull request jax-ml#9805 from froystig:lax-cleanup

  Show description for 537e35b
  

  jax authors

  committed
  537e35b

  Copy full SHA for 537e35b

• remove _dilate_shape from public jax.lax module

  froystig

  committed
  bea7771

  Copy full SHA for bea7771

• remove _ones and _zeros from public jax.lax module

  froystig

  committed
  460feed

  Copy full SHA for 460feed

• remove _delta from public jax.lax module

  froystig

  committed
  0cae316

  Copy full SHA for 0cae316

• remove _tri from public jax.lax module

  froystig

  committed
  90f31c1

  Copy full SHA for 90f31c1

• remove _eye from public jax.lax module

  froystig

  committed
  3c345ee

  Copy full SHA for 3c345ee

• remove _broadcasting_shape_rule from public jax.lax module

  froystig

  committed
  7824325

  Copy full SHA for 7824325

• remove _upcast_fp16_for_computation from public jax.lax module

  froystig

  committed
  2324e5b

  Copy full SHA for 2324e5b

• remove _check_user_dtype_supported from public jax.lax module

  froystig

  committed
  e262c72

  Copy full SHA for e262c72

• remove _reduce_or and _reduce_and from public jax.lax module

  froystig

  committed
  afc8729

  Copy full SHA for afc8729

• remove _reduce_min from public jax.lax module

  froystig

  committed
  c979f64

  Copy full SHA for c979f64

• remove _reduce_max from public jax.lax module

  froystig

  committed
  299d4db

  Copy full SHA for 299d4db

• remove _reduce_sum from public jax.lax module

  froystig

  committed
  6f51957

  Copy full SHA for 6f51957

Commits on Mar 8, 2022

• Merge pull request jax-ml#9803 from skye:create_v3_mesh

  Show description for 972070a
  

  jax authors

  committed
  972070a

  Copy full SHA for 972070a

• Improve TPU v2 and v3 mesh_utils.create_device_mesh logic.

  Show description for bcee442
  skye

  committed
  bcee442

  Copy full SHA for bcee442

• remove _one and _zero from public jax.lax module

  froystig

  committed
  7890fb7

  Copy full SHA for 7890fb7

• remove three internal functions from public jax.lax module

  Show description for 3f88518
  froystig

  committed
  3f88518

  Copy full SHA for 3f88518

• remove _eq_meet from public jax.lax module

  Show description for 7319982
  

  froystig

  authored and

  jax authors

  committed
  7319982

  Copy full SHA for 7319982

• Merge pull request jax-ml#9800 from hawkinsp:doc

  Show description for 36cd568
  

  jax authors

  committed
  36cd568

  Copy full SHA for 36cd568

• Merge pull request jax-ml#9789 from jakevdp:coo-sort-rows

  Show description for e4c8cd1
  

  jax authors

  committed
  e4c8cd1

  Copy full SHA for e4c8cd1

• Merge pull request jax-ml#9774 from jakevdp:fix-test-util

  Show description for 5b38971
  

  jax authors

  committed
  5b38971

  Copy full SHA for 5b38971

• [sparse]: COO: check for sorted rows before cusparse lowering

  jakevdp

  committed
  43c3bfd

  Copy full SHA for 43c3bfd

• [JAX] Deprecate .block_host_until_ready() in favor of .block_until_ready().

  Show description for 08fbd77
  

  hawkinsp

  authored and

  jax authors

  committed
  08fbd77

  Copy full SHA for 08fbd77

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

  Show description for 809156c
  

  jax authors

  committed
  809156c

  Copy full SHA for 809156c

• Move the construction of the in-tree up.

  Show description for aeba6b3
  

  jblespiau

  authored and

  jax authors

  committed
  aeba6b3

  Copy full SHA for aeba6b3

• remove _complex from public jax.lax module

  Show description for 3e77a56
  

  froystig

  authored and

  jax authors

  committed
  3e77a56

  Copy full SHA for 3e77a56

• Update XLA.

  hawkinsp

  committed
  81bb526

  Copy full SHA for 81bb526

Pagination

Previous

Next