- 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
Don't return tuple from Something isn't working jax linalg Linear algebra SciPy compatibility
jax.scipy.linalg.qr when mode = 'r' bug #1682 by jessegrabowski was merged Oct 20, 2025 Loading…
2 of 11 tasks
Support Scan with MIT-MOT in JAX backend backend compatibility gradients jax scan
#1651 by ricardoV94 was merged Oct 27, 2025 Loading…
Fix failing JAX Split test jax no releasenotes tests
#1646 by ricardoV94 was merged Oct 9, 2025 Loading…
Implement
wrap_jax and rename as_op to wrap_py jax major Op implementation #1614 by aseyboldt was merged Oct 2, 2025 Loading…
Pin jax temporarily as tensorflow-probability is not compatible yet GitHub CI/CD jax no releasenotes
#1567 by ricardoV94 was merged Jul 31, 2025 Loading…
Add JAX dispatch for New feature or request jax linalg Linear algebra Op implementation
CholeskySolve Op enhancement #1491 by jessegrabowski was merged Jun 21, 2025 Loading…
4 of 11 tasks
Simplify typify of Generators in JAXLinker jax no releasenotes
#1470 by ricardoV94 was merged Jun 16, 2025 Loading…
Fix nan for valid parameters in jax implementation of Multinomial bug Something isn't working jax random variables
#1328 by ricardoV94 was merged Mar 30, 2025 Loading…
Implement faster Multinomial JAX dispatch enhancement New feature or request jax random variables
#1316 by educhesne was merged Mar 25, 2025 Loading…
4 of 11 tasks
Simplify dispatch of JAX random variables by handling rng split automatically jax major random variables
#1315 by educhesne was merged Mar 24, 2025 Loading…
4 of 11 tasks
Speedup implementation of multivariate_normal and allow New feature or request jax NumPy compatibility performance random variables
method of covariance decomposition enhancement #1203 by ricardoV94 was merged Feb 13, 2025 Loading…
Implement Kve Op and Kv helper enhancement New feature or request jax Op implementation SciPy compatibility
#1081 by ricardoV94 was merged Nov 13, 2024 Loading…
Temporarily remove heaviest param of jax test_logsumexp_benchmark jax no releasenotes tests
#1061 by ricardoV94 was merged Oct 31, 2024 Loading…
Support RandomVariable graphs with scalar shape parameters in JAX backend enhancement New feature or request jax
#1029 by ricardoV94 was merged May 30, 2025 Loading…
Direct tests for dispatch implementations in alternative backends backend compatibility jax maintenance tests torch PyTorch backend
#1021 by ricardoV94 was merged Oct 10, 2024 Loading…
Reorganized JAX link folder structure jax maintenance no releasenotes
#913 by HangenYuu was merged Jul 15, 2024 Loading…
4 of 11 tasks
Implemented JAX backend for Eigvalsh backend compatibility enhancement New feature or request jax
#867 by HangenYuu was merged Jun 28, 2024 Loading…
3 of 11 tasks
Remove conservative checks for supported Subtensors operations in JAX backend compatibility indexing jax maintenance
#849 by ricardoV94 was merged Jun 28, 2024 Loading…
1 of 11 tasks
Implement JAX dispatch for Argsort and add New feature or request jax NumPy compatibility
stable argument to sorting functions enhancement #848 by ricardoV94 was merged Jun 24, 2024 Loading…
5 of 11 tasks
Add jax dispatch for
KroneckerProduct Op jax OpFromGraph #822 by jessegrabowski was merged Jun 15, 2024 Loading…
4 of 11 tasks
Fix JAX implementation of Argmax jax maintenance
#809 by HangenYuu was merged Jul 6, 2024 Loading…
4 of 11 tasks
Inline static size inputs in JAX implementation of RandomVariables jax random variables
#774 by ricardoV94 was merged May 22, 2024 Loading…
5 of 11 tasks
Fix implementation of several RandomVariables in JAX backend bug Something isn't working jax maintenance random variables
#767 by ricardoV94 was merged May 13, 2024 Loading…
5 of 9 tasks
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
Previous Next
ProTip! Filter pull requests by the default branch with base:main.