Code

[ROCm] Implement RNN support#25755

Code

Merged

copybara-service[bot] merged 1 commit intojax-ml:mainjax-ml/jax:mainfrom ROCm:ci_rnn_final-upstreamROCm/jax:ci_rnn_final-upstreamCopy head branch name to clipboardJan 15, 2025

+421-125

Commits

Commits on Jan 15, 2025

• [ROCm] Implement RNN support

  Ruturaj4

  committed
  fe68eb8

  Copy full SHA for fe68eb8