Search results

Keep jit_fn through pickle

performance

Status: Open.

#2108 In pymc-devs/pytensor;

· ricardoV94 opened on May 1, 2026

BUG: UserWarning: Numba will use object mode to run XlogY0'

enhancementNew feature or requestNew feature or request

numba

SciPy compatibility

Status: Open.

#2106 In pymc-devs/pytensor;

· aloctavodia opened on May 1, 2026

ENHANCEMENT: Scan has no MLX dispatcher — stateful recurrent logps fail to compile

enhancementNew feature or requestNew feature or request

mlx

Status: Open.

#2097 In pymc-devs/pytensor;

· cetagostini opened on Apr 28, 2026

BUG: MLX Cast dispatcher crashes when input is a Python int/float

bugSomething isn't workingSomething isn't working

mlx

Status: Open.

#2096 In pymc-devs/pytensor;

· cetagostini opened on Apr 28, 2026

PERF: minor MLX dispatch optimizations (Cast, DimShuffle, Argmax)

enhancementNew feature or requestNew feature or request

mlx

Status: Open.

#2094 In pymc-devs/pytensor;

· cetagostini opened on Apr 28, 2026

DOC: document f.trust_input=True as MLX hot-loop performance tip

docs

mlx

Status: Open.

#2093 In pymc-devs/pytensor;

· cetagostini opened on Apr 28, 2026

BUG: MLX Convolve1d dispatch crashes when Blockwise broadcasts the kernel

bugSomething isn't workingSomething isn't working

mlx

Status: Open.

#2092 In pymc-devs/pytensor;

· cetagostini opened on Apr 28, 2026

PERF: map pt.logsumexp pattern to native mx.logsumexp (1.5× speedup)

enhancementNew feature or requestNew feature or request

graph rewriting

mlx

Status: Open.

#2091 In pymc-devs/pytensor;

· cetagostini opened on Apr 28, 2026

PERF: detect scaled-dot-product-attention pattern and lower to mx.fast.scaled_dot_product_attention

enhancementNew feature or requestNew feature or request

graph rewriting

mlx

Status: Open.

#2090 In pymc-devs/pytensor;

· cetagostini opened on Apr 28, 2026

PERF: TensorType.filter is O(N) for foreign array types (mx.array, jax.Array, torch.Tensor)

backend compatibility

enhancementNew feature or requestNew feature or request

jax

mlx

Status: Open.

#2087 In pymc-devs/pytensor;

· cetagostini opened on Apr 28, 2026

BUG: MLX variadic ScalarOp crashes on broadcast-shaped composite inputs

bugSomething isn't workingSomething isn't working

mlx

Status: Open.

#2086 In pymc-devs/pytensor;

· cetagostini opened on Apr 28, 2026

MLX improvements: meta issue tracking backend perf and correctness

backend compatibility

enhancementNew feature or requestNew feature or request

mlx

Status: Open.

#2085 In pymc-devs/pytensor;

· cetagostini opened on Apr 28, 2026