Commits

Branch selector

 main

User selector

froystig

Datepicker

All time

Commit History

Commits on Oct 28, 2022

• wip bits-changing partitionable rng based on iota raveling

  Show description for 63bfb87
  

  froystig

  and

  mattjj

  committed
  63bfb87

• partitionable threefry PRNG random bits implementation

  Show description for c8b9280
  

  froystig

  and

  mattjj

  committed
  c8b9280

Commits on Oct 20, 2022

• canonicalize dtypes when loading arrays via dlpack

  froystig

  committed
  4af3509

Commits on Oct 6, 2022

• add input/output sharding to executable protocol

  froystig

  committed
  1b5e9e4

Commits on Sep 26, 2022

• add nvidia-smi question to bug template

  froystig

  committed
  2a7b319

Commits on Sep 2, 2022

• mention AOT readiness in changelog

  froystig

  committed
  a2ad414

• write in-process AOT walkthrough doc

  froystig

  committed
  bb68fbe

• write and generate package API documentation for jax.stages

  froystig

  committed
  43db064

• docstring for jax.stages.Wrapped

  froystig

  committed
  4505d57

Commits on Sep 1, 2022

• require scalar key arrays when seeding, splitting, and folding

  Show description for 5ac8043
  froystig

  committed
  5ac8043

Commits on Aug 31, 2022

• support key array pickling

  Show description for 0237643
  froystig

  committed
  0237643

• add key_data to jax.random for key array unwrapping

  Show description for 0d3630b
  froystig

  committed
  0d3630b

• test the RNG key upgrade in one of our github CI runs

  Show description for fe2de26
  froystig

  committed
  fe2de26

Commits on Aug 30, 2022

• internal rename: swap mentions of "custom eltypes" for "opaque dtypes"

  Show description for 8f045b1
  froystig

  committed
  8f045b1

• avoid unnecessary lifting to aval in jax.random.poisson

  froystig

  committed
  dc03a33

• add dtype property to key arrays

  froystig

  committed
  077bfac

• access rules through a hidden attribute of opaque dtype

  froystig

  committed
  73bf0aa

Commits on Aug 25, 2022

• minimal result-handling support for single-device key array pjit outputs

  Show description for acc025a
  

  2 peopleauthored and

  jax authors

  committed
  acc025a

• support jax.experimental.array.Array as a base array for key arrays

  Show description for 8e2d1be
  

  2 peopleauthored and

  jax authors

  committed
  8e2d1be

Commits on Aug 24, 2022

• test jax.eval_shape with key array inputs/outputs

  froystig

  committed
  af18714

• fix bug issue template yaml parsing error

  froystig

  authored
  655ecb4

Commits on Aug 22, 2022

• remove 'BUG' title prefix from bug report template

  Show description for 825cbac
  froystig

  authored
  825cbac

• roll-forward jax-ml#11952, take 2

  Show description for 6071a8f
  

  froystig

  authored and

  jax authors

  committed
  6071a8f

Commits on Aug 20, 2022

• roll-forward jax-ml#11952

  Show description for 9789e83
  

  froystig

  authored and

  jax authors

  committed
  9789e83

Commits on Aug 19, 2022

• teach jax2tf about custom eltypes, key arrays, and random key primitives

  Show description for 34b63df
  froystig

  committed
  34b63df

• enable several non-threefry RNG tests without config.jax_enable_custom_prng

  froystig

  committed
  82243d0

• add key array checks on entry to random.{orthogonal,generalized_normal,ball}

  froystig

  committed
  6939edd

• defer to custom eltype for sharded result handling, use this to handle sharded key arrays

  froystig

  committed
  affb031

• introduce key-element-type arrays and overhaul the Python PRNG key array type

  Show description for 7f06df1
  froystig

  committed
  7f06df1

• sketch: setup for new key array implementation based on eltypes

  Show description for acb5e49
  

  froystig

  and

  mattjj

  committed
  acb5e49

Commits on Aug 12, 2022

• defer to custom eltype for gather lowering rule

  Show description for c9d3094
  froystig

  committed
  c9d3094

• skip redundant asarray conversion in LAX numpy rewriting-take helper

  froystig

  committed
  91b90f4

Commits on Aug 10, 2022

• convert custom VJP update guide to a retroactive JEP

  froystig

  committed
  c6ab3a6

• update checkpoint attributes according to functools.wraps

  Show description for 7d494a3
  froystig

  committed
  7d494a3

• defer to custom eltype for slice lowering rule

  Show description for 7955799
  froystig

  committed
  7955799

Pagination

Previous

Next