Skip to content

Issues: pymc-devs/pytensor

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

Clear current search query, filters, and sorts
14 Open 77 Closed
14 Open 77 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Assignee
Filter by who’s assigned
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Issues list

Transform graph to make large constants symbolic inputs enhancement New feature or request jax
#1224 opened Feb 19, 2025 by ricardoV94
1
Implement @as_jax_op to wrap a JAX function for use in PyTensor enhancement New feature or request jax Op implementation
#1120 opened Dec 12, 2024 by jdehning • Draft
5 of 14 tasks
1
66
Two parametrization of JAX benchmark test are taking hours jax needs info Additional information required tests
#1060 opened Oct 31, 2024 by ricardoV94
2
Support RandomVariable graphs with scalar shape parameters in JAX backend enhancement New feature or request jax
#1029 opened Oct 11, 2024 by ricardoV94 Loading…
2
Use expand_dims / squeeze in JAX implementation of Dimshuffle beginner friendly jax
#847 opened Jun 24, 2024 by ricardoV94
1
2
minor bug: jax tests reference jax.interpreters.xla.DeviceArray beginner friendly bug Something isn't working jax tests
#660 opened Mar 7, 2024 by leventov
2
Implement helper @as_jax_op to wrap JAX functions in PyTensor backend compatibility feature request jax torch PyTorch backend
#537 opened Dec 7, 2023 by ricardoV94
1
4
No JAX dispatch for mul_without_zeros backend compatibility enhancement New feature or request jax
#526 opened Nov 30, 2023 by jessegrabowski
9
Implement proper JAX/Numba dispatch for IfElse backend compatibility jax numba
#501 opened Nov 15, 2023 by ricardoV94
4
BUG: Pytensor-JAX unable to use dynamic slice indexing backend compatibility feature request jax
#312 opened May 19, 2023 by pbaggens1
Implement JAX and Numba dispatchers for ScalarLoop backend compatibility help wanted Extra attention is needed jax numba
#299 opened May 14, 2023 by ricardoV94
Allow explicit RNG and Sparse input types in JAX functions backend compatibility enhancement New feature or request jax random variables sparse variables
#278 opened Apr 18, 2023 by ricardoV94 Loading…
30
Implement JAX conversion for While Scan backend compatibility help wanted Extra attention is needed jax scan
#265 opened Apr 5, 2023 by ricardoV94
Implement JAX conversion for Scan with mit-mot backend compatibility help wanted Extra attention is needed jax scan
#264 opened Apr 5, 2023 by ricardoV94
2
ProTip! Updated in the last three days: updated:>2025-04-26.