mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-26 17:26:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
14,449 Commits 508 Branches 263 Tags
Commit Graph

224 Commits

Author SHA1 Message Date
Matthew Johnson
376ee6423f
Merge pull request #631 from fehiepsi/bernoulli 
Implement bernoulli logpmf
 2019-04-23 16:59:40 -07:00
fehiepsi
a790436be4 add dirichlet sampler 2019-04-22 11:55:02 -04:00
fehiepsi
665e72a23a implement bernoulli pmf 2019-04-21 21:22:50 -04:00
fehiepsi
088081711d add tests 2019-04-21 16:43:18 -04:00
fehiepsi
e5ccf0534d add beta/t random samplers 2019-04-21 16:25:20 -04:00
Matthew Johnson
c5a381ed4d misc test skips 2019-04-15 09:14:43 -07:00
Matthew Johnson
35e340670c
Merge branch 'master' into gamma 2019-04-12 07:15:41 -07:00
Jamie Townsend
7b13ae4b41 Add Gumbel to jax.random, and test 2019-04-02 10:55:03 +01:00
fehiepsi
80e22d2fdf Merge remote-tracking branch 'upstream/master' into gamma 2019-03-31 23:54:46 -04:00
fehiepsi
e0567b6d16 add Gamma sampler 2019-03-30 18:07:34 -04:00
fehiepsi
096ba63ba6 correct TODO test name for bernoulli 2019-03-30 16:45:17 -04:00
fehiepsi
bb095d3df5 implement pareto logpdf and sampler 2019-03-30 16:34:20 -04:00
fehiepsi
890ba842a9 implement expon and laplace sampler 2019-03-28 23:57:00 -04:00
fehiepsi
a0636eaedd implement cauchy pdf and random sampler 2019-03-28 17:59:42 -04:00
Matthew Johnson
0e749f29ef use unittest.SkipTest to skip tests properly 2019-03-19 16:38:42 -07:00
fehiepsi
f05d0bcbd8 fix bernoulli shape bug 2019-03-02 19:11:16 -05:00
Matthew Johnson
0ff98a74eb add random.fold_in, update mnist_vae.py loops 2019-02-13 09:55:36 -08:00
Matthew Johnson
70d1a00443 set behavior when random.randint has invalid range 
(closes #222)
 2019-01-12 13:12:40 -08:00
Matthew Johnson
d3ec0b23c5 fix miscellaneous bugs, incl. complex abs grad 2018-12-12 12:03:57 -05:00
Dougal Maclaurin
bb8899f60d Updated existing parameterized tests to subsample 2018-12-06 18:37:59 -05:00
Dougal Maclaurin
2df36f7510 Made a shim to handle configuration without having absl parse command-line flags. 
PiperOrigin-RevId: 223391288
 2018-11-29 13:44:54 -08:00
Peter Hawkins
e180f08113 source sync 
PiperOrigin-RevId: 222451919
 2018-11-21 20:22:51 -08:00
Matthew Johnson
46c6a9170f sync updates 2018-11-19 07:47:59 -08:00
Matthew Johnson
489dd1c81a move jax tests and examples to repo root 2018-11-18 15:15:47 -08:00