memorax
A unified JAX/Flax framework for memory-augmented reinforcement learning with support for RNNs, SSMs, and Transformers
Activity
- Latest release
- 7mo ago
- Total releases
- 3
- Cadence
- ~2 months
- Last 12 months
- 3
Details
- License
- Apache-2.0
- First release
- Oct 21, 2025
Releases
| Version | Released | |
|---|---|---|
1.0.1
patch
|
1.0.1
patch
Dependencies (19)
+ 11 more |
|
1.0.0
major
|
1.0.0
major
Dependencies (18)
+ 10 more |
|
0.1.0
initial
|
0.1.0
initial
|