Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add JAX implementation for LogNormalRV #1346

Merged
merged 1 commit into from
Dec 10, 2022

Conversation

juanitorduz
Copy link
Contributor

[
set_test_value(
at.lvector(),
np.array([0, 0], dtype=np.int64),
Copy link
Contributor Author

@juanitorduz juanitorduz Dec 9, 2022

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Seem scipy's implementation does not accept a loc param ? https://docs.scipy.org/doc/scipy/reference/generated/scipy.stats.lognorm.html

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

It seems to work with something like scipy.stats.lognorm(1, loc=10).rvs().

@brandonwillard brandonwillard requested a review from rlouf December 9, 2022 20:12
@brandonwillard brandonwillard added JAX Involves JAX transpilation random variables Involves random variables and/or sampling labels Dec 9, 2022
@rlouf rlouf changed the title jax lognormal Add JAX implementation for LogNormalRV Dec 9, 2022
@codecov
Copy link

codecov bot commented Dec 9, 2022

Codecov Report

Merging #1346 (8d3089d) into main (fe3e76d) will increase coverage by 0.00%.
The diff coverage is 100.00%.

Additional details and impacted files

Impacted file tree graph

@@           Coverage Diff           @@
##             main    #1346   +/-   ##
=======================================
  Coverage   74.35%   74.36%           
=======================================
  Files         177      177           
  Lines       49036    49046   +10     
  Branches    10379    10379           
=======================================
+ Hits        36461    36471   +10     
  Misses      10283    10283           
  Partials     2292     2292           
Impacted Files Coverage Δ
aesara/link/jax/dispatch/random.py 100.00% <100.00%> (ø)

@rlouf
Copy link
Member

rlouf commented Dec 10, 2022

Thank you for contributing!

@rlouf rlouf merged commit 2224697 into aesara-devs:main Dec 10, 2022
@juanitorduz juanitorduz deleted the jax_lognormal branch December 10, 2022 11:17
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
JAX Involves JAX transpilation random variables Involves random variables and/or sampling
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Add JAX implementation for LogNormalRV
3 participants