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.

Sign up for GitHub

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

Jump to bottom

Full JAX coverage #1425

Open
5 of 20 tasks
rlouf opened this issue Feb 13, 2023 · 0 comments
Open
5 of 20 tasks

Full JAX coverage #1425

rlouf opened this issue Feb 13, 2023 · 0 comments
Labels
documentation Improvements or additions to documentation

Comments

@rlouf
Copy link
Member

rlouf commented Feb 13, 2023 •
edited

Distributions
Beta Give feedback
  1. Add JAX implementation for InvGammaRV #1368
    JAX enhancement good first issue help wanted random variables
  2. Add JAX implementation for GenGammaRV #1333
    JAX enhancement good first issue help wanted random variables
  3. Add JAX implementation for BetaBinomialRV #1332
    JAX enhancement good first issue help wanted random variables
  4. Add JAX implementation for IntegersRV #1331
    JAX enhancement good first issue help wanted random variables
  5. Add JAX implementation for VonMissesRV #1330
    JAX good first issue help wanted random variables
  6. Add JAX implementation for NegBinomialRV #1327
    JAX enhancement good first issue help wanted random variables
  7. Add JAX implementation for WaldRV #1329
    JAX enhancement good first issue help wanted random variables
  8. Add JAX implementation for MultinomialRV #1326
    JAX enhancement good first issue help wanted random variables
  9. Add JAX implementation for HypergeometricRV #1324
    JAX enhancement good first issue help wanted random variables
  10. Add JAX implementation for GeometricRV #1323
    JAX enhancement good first issue help wanted random variables
  11. Add JAX implementation for ChiSquareRV #1322
    JAX enhancement good first issue help wanted random variables
  12. Add JAX implementation for BinomialRV #1321
    JAX enhancement good first issue help wanted random variables

User interface
Beta Give feedback
  1. The function returned by aesara.function(..., mode="JAX") is not compatible with JAX out of the box #1194
    JAX enhancement
    rlouf
  2. Improve the interface to JAX-compiled functions #1385
    JAX enhancement

Scan
Beta Give feedback
  1. JAX backend fails with simple Scan examples #924
    JAX Scan bug help wanted
    rlouf
  2. Implement mit-mots conversion in the JAX translation of the Scan Op #186
    JAX Scan enhancement help wanted
  3. Implement Scan's as_while in JAX #710
    JAX Scan bug help wanted important

Op coverage
Beta Give feedback
  1. Limited implementation of ARange Op in JAX #68
    JAX help wanted
  2. Apply JAX JIT more selectively #684
    JAX question request discussion
  3. Jax 0.2 does not support jax.numpy.reshape with non-constant values in omnistaging mode #43
    JAX enhancement question
@rlouf rlouf added the documentation Improvements or additions to documentation label Feb 13, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees
No one assigned
Labels
documentation Improvements or additions to documentation
Projects
Aesara Roadmap
Status: Backends
Milestone
No milestone
Development

No branches or pull requests
1 participant
@rlouf