8000 Adding `ops.associative_scan`? · Issue #19904 · keras-team/keras · GitHub
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content
Adding ops.associative_scan? #19904
New issue
New issue
Closed
Closed
Adding ops.associative_scan?#19904
Assignees
mehtamansi29
@SeKim12

Description

@SeKim12
SeKim12
opened on Jun 22, 2024

Hello!

I was looking for the equivalent of jax.lax.associative_scan. This is an important operation for recent state space models, e.g. S5, Mamba.

Maybe it's there and I'm missing it. Otherwise, I could start working on it. Thanks!

Metadata

Metadata

Assignees

Labels

No labels
No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions
    0