Issues

• Expose compile_train's update assembly without compiling

  enhancementoptimizer
  pymc-devs/pytensor-ml#168 · Filter by author jessegrabowski opened on Oct 2, 2026

• scaled_dot_product_attention promotes float32 inputs to float64 (uses config.floatX for the scale and causal mask)

  pymc-devs/pytensor-ml#164 · Filter by author cetagostini opened on Sep 19, 2026

• Guard the MLX convolution's rank at compile time

  enhancementpytensor
  pymc-devs/pytensor-ml#135 · Filter by author jessegrabowski opened on Aug 28, 2026
  1

• Add a batch_first argument to MultiheadAttention

  enhancementlayer
  pymc-devs/pytensor-ml#127 · Filter by author jessegrabowski opened on Aug 28, 2026
  1

• require_unique_state_names misses a duplicate name that is only read

  enhancementoptimizer
  pymc-devs/pytensor-ml#122 · Filter by author jessegrabowski opened on Aug 28, 2026
  1

• JAX aligns the causal mask top-left, so generation with a KV cache attends to the wrong prefix

  buglayer
  pymc-devs/pytensor-ml#120 · Filter by author jessegrabowski opened on Aug 28, 2026

• Test groups have drifted out of balance, so CI wall clock is 8 minutes

  enhancement
  pymc-devs/pytensor-ml#109 · Filter by author jessegrabowski opened on Aug 24, 2026

• compile_train gives no way to get the shared variables its step writes

  enhancement
  pymc-devs/pytensor-ml#80 · Filter by author jessegrabowski opened on Aug 10, 2026

• Implement SparseAdam optimizer

  pymc-devs/pytensor-ml#59 · Filter by author jessegrabowski opened on Aug 8, 2026

• Implement ASGD optimizer

  pymc-devs/pytensor-ml#57 · Filter by author jessegrabowski opened on Aug 8, 2026

• Implement LBFGS optimizer

  pymc-devs/pytensor-ml#58 · Filter by author jessegrabowski opened on Aug 8, 2026

• Implement RAdam optimizer

  pymc-devs/pytensor-ml#56 · Filter by author jessegrabowski opened on Aug 8, 2026

• Meta: native LLM inference runtime in pytensor_ml (tracking)

  pymc-devs/pytensor-ml#53 · Filter by author cetagostini opened on Aug 4, 2026
  2

• Create an example gallery

  pymc-devs/pytensor-ml#4 · Filter by author jessegrabowski opened on May 3, 2024