carlosgmartin
|
6b69a136aa
|
Add jax.random.multinomial.
|
2025-03-12 18:15:14 -04:00 |
|
Jake VanderPlas
|
e4dac395a5
|
Roll back multinomial change from https://github.com/jax-ml/jax/pull/25688
This has test breakages on TPU: https://github.com/jax-ml/jax/actions/runs/13159081976/job/36723019653
Reverts 95535df13b422284043623ca3a6d2a5962116fb1
PiperOrigin-RevId: 723536107
|
2025-02-05 09:13:56 -08:00 |
|
carlosgmartin
|
32411a430f
|
Add jax.random.multinomial.
|
2025-01-31 18:45:55 -05:00 |
|
Roy Frostig
|
dd535d88a7
|
emphasize typed over legacy RNG keys in random module docs
Update both docstrings and move the `PRNGKey` function listing lower
in the API reference.
|
2024-08-11 12:41:50 -07:00 |
|
Jake VanderPlas
|
6771a59181
|
[key reuse] add jax.random.clone
|
2024-03-08 09:06:00 -08:00 |
|
jiayaobo
|
ae2387dc27
|
add random.binomial
update
update
modify
|
2023-11-19 14:51:10 +08:00 |
|
Jake VanderPlas
|
5f7335fb55
|
Deprecate jax.random.shuffle
This has been long deprecated, but this PR uses the standard deprecation
framework to make it easier to finalize.
|
2023-11-06 12:21:56 -08:00 |
|
Jake VanderPlas
|
406bfc54c2
|
DOC: reorganize jax.random API listing
|
2023-09-21 11:13:23 -07:00 |
|
Roy Frostig
|
2bf9322ccc
|
move wrap_key_data to jax.random
This is a fine function for the public API, rather than `jax.extend`.
|
2023-09-18 14:38:22 -07:00 |
|
Jake VanderPlas
|
f935c00643
|
DOC: add missing docs for jax.random functions
|
2023-08-29 11:20:25 -07:00 |
|
jiayaobo
|
6d184459ef
|
add random.triangular and random.lognormal
add random.triangular and random.lognormal
add random.triangular and random.lognormal
|
2023-08-05 12:27:53 +08:00 |
|
Roy Frostig
|
ea3389205f
|
add jax.random.bits
|
2023-05-03 06:10:05 -07:00 |
|
Matthew Johnson
|
a7f5e07549
|
update prng docs to mention jax_threefry_partitionable
fixes #15484
|
2023-04-07 22:55:15 -07:00 |
|
jiayaobo
|
924894e85c
|
add geometric random gen
add geom random
add geom random
add geom random
add geom random
|
2023-03-30 02:08:04 +08:00 |
|
jiayaobo
|
f7a14d65d2
|
add wald random generator
add wald to random.py
|
2023-03-22 11:06:59 +08:00 |
|
jiayaobo
|
05c47033b2
|
add rayleigh distribution to random.py
add rayleigh distribution to random.py
add rayleigh distribution to random.py
add rayleigh to random.py
|
2023-03-14 09:55:54 +08:00 |
|
jiayaobo
|
fdf8ac18d6
|
add random.chisquare and random.f
add chi2 and F random variables methods
add chi2 and F random variables methods
fix F rv shape broadcasting
fix shape broadcasting
|
2023-03-01 15:03:50 +08:00 |
|
Jake VanderPlas
|
81e627d5bd
|
DOC: make API doc titles more uniform
|
2023-01-18 10:59:42 -08:00 |
|
carlosgmartin
|
ca83a80f95
|
Added random.generalized_normal and random.ball.
|
2022-06-03 15:11:29 -04:00 |
|
Carlos Martin
|
b276c31b75
|
Added random.orthogonal.
|
2022-04-29 14:20:50 -04:00 |
|
Jake VanderPlas
|
69969ef803
|
add random.loggamma and improve dirichlet & beta implementation
|
2022-03-21 08:33:11 -07:00 |
|
Jake VanderPlas
|
4ae393f8a8
|
DOC: fix jax.random documentation
|
2020-11-20 16:41:32 -08:00 |
|
George Necula
|
4e89d43a75
|
Added JAX pytrees notebook
Also added docstrings to the tree_util module.
|
2019-11-24 20:29:07 +01:00 |
|
Matthew Johnson
|
c467bbfe50
|
add jax.random to docs
|
2019-02-13 19:21:55 -08:00 |
|