Search results

Compatibility issue between <code>add_loss</code> and <code>TimeDistributed</code> when using JAX backend

backend:jax

stat:awaiting keras-engAwaiting response from Keras engineerAwaiting response from Keras engineer

type:Bug

Status: Open.

#21605 In keras-team/keras;

· AmedeoBiolatti opened on Aug 21, 2025

custom_gradient not working with JAX backend

backend:jax

type:Bug

Status: Open.

#21105 In keras-team/keras;

· larschristensen opened on Mar 28, 2025

Use JAX's AbstractMesh in distribution lib

backend:jax

type:performance

Status: Open.

#21027 In keras-team/keras;

· hertschuh opened on Mar 13, 2025

Unable to see any speedup from multi-GPU training using JAX backend for LSTM / GRU

backend:jax

Status: Open.

#20998 In keras-team/keras;

· larschristensen opened on Mar 6, 2025

Request for Assistance with JAX Backend: Custom Model Training with Multiple Submodels

backend:jax

stat:awaiting keras-engAwaiting response from Keras engineerAwaiting response from Keras engineer

type:supportUser is asking for help / asking an implementation question. Stackoverflow would be better suited.User is asking for help / asking an implementation question. Stackoverflow would be better suited.

Status: Open.

#19358 In keras-team/keras;

· abhaskumarsinha opened on Mar 22, 2024

XlaRuntimeError for TopK metrics with JAX backend

backend:jax

stat:awaiting keras-engAwaiting response from Keras engineerAwaiting response from Keras engineer

type:Bug

Status: Open.

#19290 In keras-team/keras;

· innat opened on Mar 12, 2024

ExportArchive cannot be used with JAX

backend:jax

stat:awaiting keras-engAwaiting response from Keras engineerAwaiting response from Keras engineer

type:Bug

Status: Open.

#19134 In keras-team/keras;

· martin-gorner opened on Feb 1, 2024