Code

support custom PRNG implementations#6899

Code

Merged

copybara-service[bot] merged 6 commits intomainjax-ml/jax:mainfrom custom-rngjax-ml/jax:custom-rngCopy head branch name to clipboardAug 20, 2021

+998-542

Commits

Commits on Aug 18, 2021

• factor PRNG routines from random module to prng

  froystig

  committed
  b4ccecc

Commits on Aug 20, 2021

• introduce custom PRNG implementations and an array-like adapter for them

  Show description for aa265cc
  froystig

  committed
  aa265cc

• factor prng tests from lax random tests

  Show description for e33ef7f
  froystig

  committed
  e33ef7f

• introduce custom PRNG tests

  Show description for ad4b454
  froystig

  committed
  ad4b454

• implement backwards-compatible behavior and enable custom PRNGs only conditionally

  Show description for 60e0e9f
  froystig

  committed
  60e0e9f

• alias prng.threefry_2x32 in random and warn of move

  Show description for 4eb437a
  froystig

  committed
  4eb437a