Search Packages

fla-core

🚀 Efficient implementations for emerging model architectures

flash-linear-attention

🚀 Efficient implementations for emerging model architectures

gateloop-transformer

Implementation of GateLoop Transformer in Pytorch and Jax

rwkv-fla

🚀 Efficient implementations for emerging model architectures

decepticons

A backend-neutral kernel of predictive primitives for descendant systems. O(n) attention is deception.

arabicnlp

Python package for Arabic natural language processing

s5-pytorch

Pytorch implementation of Simplified Structured State-Spaces for Sequence Modeling (S5)

pyxlstm

Efficient Python library for Extended LSTM with exponential gating, memory mixing, and matrix memory for superior sequence modeling.

discretax

State Space Models in Jax

transact-tf

An implementation of TransAct: Transformer-based Realtime User Action Model for Recommendation

linax

State Space Models in Jax

lrnnx

A unified PyTorch library for Linear RNNs

spectrans

Modular spectral transformer implementations in PyTorch