- Notifications
You must be signed in to change notification settings - Fork 149
Pull requests: pymc-devs/pytensor
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Add New feature or request help wanted Extra attention is needed jax NumPy compatibility Op implementation
einsum enhancement #722 by jessegrabowski was merged Aug 4, 2024 Loading…
13 of 21 tasks
Add support for random Generators in Numba backend bug Something isn't working enhancement New feature or request jax major numba Op implementation random variables
#691 by ricardoV94 was merged May 29, 2024 Loading…
17 of 19 tasks
Add JAX support for SortOp backend compatibility enhancement New feature or request jax
#657 by HarshvirSandhu was merged Mar 4, 2024 Loading…
4 tasks done
Allow more dynamic shape scans to be compiled enhancement New feature or request jax scan shape inference
#647 by ricardoV94 was merged Mar 13, 2024 Loading…
6 of 11 tasks
Fix bug in storage_input alignment of the JAX backend bug Something isn't working jax random variables
#587 by ricardoV94 was merged Jan 12, 2024 Loading…
6 of 10 tasks
Implement JAX dispatch for TriGamma backend compatibility enhancement New feature or request jax
#498 by ricardoV94 was merged Nov 15, 2023 Loading…
Defer the use of nfunc_spec in JAX scalar dispatch jax maintenance
#496 by ferrine was merged Nov 14, 2023 Loading…
Support Blockwise in JAX backend backend compatibility enhancement New feature or request jax
#487 by ricardoV94 was merged Nov 10, 2023 Loading…
Use boolean Something isn't working jax
__props__ in SVD Op bug #467 by ricardoV94 was merged Oct 4, 2023 Loading…
Do not invert Gamma rate and fix JAX implementation of Gamma and Pareto bug Something isn't working jax major Op implementation random variables
#460 by tvwenger was merged Sep 29, 2023 Loading…
6 tasks done
Exclude backend incompatible rewrites in Scan dispatch backend compatibility bug Something isn't working graph rewriting jax numba scan
#427 by ricardoV94 was merged Sep 17, 2023 Loading…
Provide JAX Ops from Optional tensorflow-probability dependency enhancement New feature or request jax
#403 by ricardoV94 was merged Jul 27, 2023 Loading…
Don't include Something isn't working graph rewriting jax
local_uint_constant_indices rewrite in JAX mode due to XLA bug bug #400 by ricardoV94 was merged Jul 26, 2023 Loading…
Extend Something isn't working enhancement New feature or request gradients jax numba
ExtractDiag gradient and numba implementation to higher dimensional inputs backend compatibility bug #389 by ricardoV94 was merged Aug 24, 2023 Loading…
Fix bug in JAX implementation of Second bug Something isn't working jax
#374 by ricardoV94 was merged Jul 11, 2023 Loading…
Forbid runtime broadcasting in Elemwise backend compatibility C-backend jax major numba Op implementation shape inference shape problem
#372 by ricardoV94 was merged Jul 12, 2023 Loading…
Don't run useless fusion and inplace rewrites in JAX mode graph rewriting jax maintenance
#370 by ricardoV94 was merged Jul 4, 2023 Loading…
Cleanup fusion rewrite database graph rewriting jax maintenance
#369 by ricardoV94 was merged Jul 4, 2023 Loading…
Fix failing test due to change in JAX behavior jax maintenance no releasenotes tests
#362 by ricardoV94 was merged Jun 30, 2023 Loading…
Fix bug in Something isn't working graph rewriting jax
boolean_indexing_set_or_inc bug #359 by ricardoV94 was merged Jun 26, 2023 Loading…
Fix bug in JAX cloning of RNG shared variables bug Something isn't working jax tests
#315 by ricardoV94 was merged May 24, 2023 Loading…
Add JAX support for New feature or request jax
pt.tri backend compatibility enhancement #302 by jessegrabowski was merged May 18, 2023 Loading…
6 tasks
Fix pinv jax test and support hermitian flag backend compatibility enhancement New feature or request jax no releasenotes tests
#296 by ricardoV94 was merged May 14, 2023 Loading…
Add jax implementation of New feature or request jax NumPy compatibility
pt.linalg.pinv backend compatibility enhancement #294 by jessegrabowski was merged May 13, 2023 Loading…
2 tasks
ProTip! Type g i on any issue or pull request to go back to the issue listing page.