Search results

Transform graph to make large constants symbolic inputs

enhancementNew feature or requestNew feature or request

jax

Status: Open.

#1224 In pymc-devs/pytensor;

· ricardoV94 opened on Feb 19, 2025

Two parametrization of JAX benchmark test are taking hours

jax

needs infoAdditional information requiredAdditional information required

tests

Status: Open.

#1060 In pymc-devs/pytensor;

· ricardoV94 opened on Oct 31, 2024

Use <code>expand_dims</code> / <code>squeeze</code> in JAX implementation of <code>Dimshuffle</code>

beginner friendly

jax

Status: Open.

#847 In pymc-devs/pytensor;

· ricardoV94 opened on Jun 24, 2024

minor bug: jax tests reference jax.interpreters.xla.DeviceArray

beginner friendly

bugSomething isn't workingSomething isn't working

jax

tests

Status: Open.

#660 In pymc-devs/pytensor;

· leventov opened on Mar 7, 2024

Implement helper <code>@as_jax_op</code> to wrap JAX functions in PyTensor

backend compatibility

feature request

jax

torchPyTorch backendPyTorch backend

Status: Open.

#537 In pymc-devs/pytensor;

· ricardoV94 opened on Dec 7, 2023

No JAX dispatch for <code>mul_without_zeros</code>

backend compatibility

enhancementNew feature or requestNew feature or request

jax

Status: Open.

#526 In pymc-devs/pytensor;

· jessegrabowski opened on Nov 30, 2023

Implement proper JAX/Numba dispatch for IfElse

backend compatibility

jax

numba

Status: Open.

#501 In pymc-devs/pytensor;

· ricardoV94 opened on Nov 15, 2023

BUG: Pytensor-JAX unable to use dynamic slice indexing

backend compatibility

feature request

jax

Status: Open.

#312 In pymc-devs/pytensor;

· pbaggens1 opened on May 19, 2023

Implement JAX and Numba dispatchers for ScalarLoop

backend compatibility

help wantedExtra attention is neededExtra attention is needed

jax

numba

Status: Open.

#299 In pymc-devs/pytensor;

· ricardoV94 opened on May 14, 2023

Implement JAX conversion for While Scan

backend compatibility

help wantedExtra attention is neededExtra attention is needed

jax

scan

Status: Open.

#265 In pymc-devs/pytensor;

· ricardoV94 opened on Apr 5, 2023

Implement JAX conversion for Scan with mit-mot

backend compatibility

help wantedExtra attention is neededExtra attention is needed

jax

scan

Status: Open.

#264 In pymc-devs/pytensor;

· ricardoV94 opened on Apr 5, 2023