Search results

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: detect RMSNorm pattern and lower to mx.fast.rms_norm (1.7× speedup)

enhancementNew feature or requestNew feature or request

graph rewriting

mlx

Status: Open.

#2089 In pymc-devs/pytensor;

· cetagostini opened on Apr 28, 2026

PERF: detect LayerNorm pattern and lower to mx.fast.layer_norm (2.4× speedup)

enhancementNew feature or requestNew feature or request

graph rewriting

mlx

Status: Open.

#2088 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