Commits

Branch selector

 main

User selector

froystig

Datepicker

All time

Commit History

Commits on Jun 7, 2024

• rename Specialized to Traced (and specialize to trace)

  Show description for ea6dfd1
  

  froystig

  authored and

  jax authors

  committed
  ea6dfd1

  Copy full SHA for ea6dfd1

Commits on Jun 6, 2024

• support both static- and dynamic-shaped arguments to djited functions

  Show description for 9db8c6a
  

  froystig

  and

  mattjj

  committed
  9db8c6a

  Copy full SHA for 9db8c6a

Commits on Jun 5, 2024

• configure dynamic shape mode only locally

  Show description for 6b275a1
  froystig

  committed
  6b275a1

  Copy full SHA for 6b275a1

• remove usage of eval_shape when defining allocating ops

  Show description for 1aba5c2
  froystig

  committed
  1aba5c2

  Copy full SHA for 1aba5c2

• add helpers for internal static shape/dimension assertions

  Show description for 971afab
  froystig

  committed
  971afab

  Copy full SHA for 971afab

• introduce a dynamic-shape jax frontend that lowers via slabby jax

  Show description for 5be5352
  

  froystig

  and

  mattjj

  committed
  5be5352

  Copy full SHA for 5be5352

• implement matmul!

  Show description for b823aa7
  

  froystig

  and

  apaszke

  committed
  b823aa7

  Copy full SHA for b823aa7

• reintroduce elementwise operations

  Show description for 3732d30
  

  froystig

  and

  apaszke

  committed
  3732d30

  Copy full SHA for 3732d30

• rewrite the slab in the units that we (think we) want

  Show description for 9651264
  

  froystig

  and

  apaszke

  committed
  9651264

  Copy full SHA for 9651264

• extend elementwise operations to multi-dimensional views

  Show description for 1cd6a7d
  

  froystig

  and

  mattjj

  committed
  1cd6a7d

  Copy full SHA for 1cd6a7d

• new slab layout

  Show description for eb68d68
  

  3 peoplecommitted
  eb68d68

  Copy full SHA for eb68d68

• heap/slab dynamic shape machine experiment

  Show description for b5b55db
  froystig

  committed
  b5b55db

  Copy full SHA for b5b55db

Commits on May 1, 2024

• reintroduce the Threefry GPU kernel lowering, under a flag

  Show description for 3f95407
  

  froystig

  authored and

  jax authors

  committed
  3f95407

  Copy full SHA for 3f95407

Commits on Apr 30, 2024

• bump shard count for random_lax_test

  Show description for c18f791
  

  froystig

  authored and

  jax authors

  committed
  c18f791

  Copy full SHA for c18f791

• remove Threefry GPU kernel

  Show description for 69878c4
  

  froystig

  authored and

  jax authors

  committed
  69878c4

  Copy full SHA for 69878c4

Commits on Apr 13, 2024

• fix up extend:core build rule

  Show description for 0941560
  

  froystig

  authored and

  jax authors

  committed
  0941560

  Copy full SHA for 0941560

Commits on Apr 10, 2024

• add and populate jax.extend.core.primitives

  froystig

  committed
  65034b3

  Copy full SHA for 65034b3

Commits on Apr 4, 2024

• changelog: note doc change to use jax.random.key over PRNGKey

  froystig

  committed
  f247822

  Copy full SHA for f247822

• changelog: batching rule change for rng_bit_generator

  froystig

  committed
  2a36d75

  Copy full SHA for 2a36d75

Commits on Mar 10, 2024

• set a lower pval for vmap-of-rbg based uniform statistical test

  Show description for 477a5aa
  

  froystig

  authored and

  jax authors

  committed
  477a5aa

  Copy full SHA for 477a5aa

Commits on Mar 8, 2024

• define a loop-free untrue batching rule for rng_bit_generator

  froystig

  committed
  29edfd8

  Copy full SHA for 29edfd8

Commits on Mar 7, 2024

• add key array upgrade note to jax.random module doc

  froystig

  committed
  721ca3f

  Copy full SHA for 721ca3f

• update pallas quickstart to new-style typed PRNG keys

  froystig

  committed
  fe3e798

  Copy full SHA for fe3e798

• update jax-101 to new-style typed PRNG keys

  froystig

  committed
  0fb4085

  Copy full SHA for 0fb4085

• update top-level examples to use new-style typed keys

  froystig

  committed
  78fd4f1

  Copy full SHA for 78fd4f1

• update experimental.sparse to new-style typed keys

  froystig

  committed
  753a6b2

  Copy full SHA for 753a6b2

• update stax to new-style typed keys

  froystig

  committed
  7fe44c1

  Copy full SHA for 7fe44c1

• update notebooks to new-style typed PRNG keys

  froystig

  committed
  75a53f4

  Copy full SHA for 75a53f4

• update documentation notes to new-style typed keys

  froystig

  committed
  0bdbe76

  Copy full SHA for 0bdbe76

• update package/API reference docs to new-style typed PRNG keys

  froystig

  committed
  98f790f

  Copy full SHA for 98f790f

Commits on Mar 6, 2024

• remove dead code in random

  froystig

  committed
  6240dc6

  Copy full SHA for 6240dc6

Commits on Jan 29, 2024

• remove PRNGKeyArray ABC

  Show description for a043325
  

  froystig

  authored and

  jax authors

  committed
  a043325

  Copy full SHA for a043325

Commits on Jan 26, 2024

• remove key array's isinstance-overriding metaclass

  Show description for 2478f31
  

  froystig

  authored and

  jax authors

  committed
  2478f31

  Copy full SHA for 2478f31

Commits on Dec 19, 2023

• decrease sample size of dirichlet test

  Show description for 66e86a3
  

  froystig

  authored and

  jax authors

  committed
  66e86a3

  Copy full SHA for 66e86a3

Commits on Dec 14, 2023

• split the random generalized normal test and skip its K-S half

  Show description for 3380b9f
  

  froystig

  authored and

  jax authors

  committed
  3380b9f

  Copy full SHA for 3380b9f

Pagination

Previous

Next