A miscellaneous set of helper functions, custom distributions, and other utilities that I find useful when using NumPyro in my work
APACHE-2.0 License
Bot releases are hidden (Show)
forward_mode_differentiation
kwarg to JaxOptimize by @vandalt in https://github.com/dfm/numpyro-ext/pull/32
Full Changelog: https://github.com/dfm/numpyro-ext/compare/v0.0.3...v0.0.4
Published by dfm 8 months ago
conditional
interface to marginalized linear distribution by @dfm in https://github.com/dfm/numpyro-ext/pull/17
Full Changelog: https://github.com/dfm/numpyro-ext/compare/v0.0.2...v0.0.3
Published by dfm 8 months ago
Published by dfm 8 months ago
Published by dfm over 1 year ago
Full Changelog: https://github.com/dfm/numpyro-ext/compare/v0.0.1...v0.0.2
Published by dfm over 1 year ago
Full Changelog: https://github.com/dfm/numpyro-ext/compare/v0.0.1...v0.0.2rc1
Published by dfm about 2 years ago
Initial release