Giters

aesara-devs / aesara

Aesara is a Python library for defining, optimizing, and efficiently evaluating mathematical expressions involving multi-dimensional arrays.

Home Page:https://aesara.readthedocs.io

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Full JAX coverage

rlouf opened this issue · comments

commented

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