A pure pytorch implementation of xLSTM.
Project description
xLSTM
A pure pytorch implementation of the XLSTM paper.
TODO
- Ensure correct with paper
- Create some usage examples
- Implement paralleization
- CUDA?
- Allow for different initializations according to: https://pytorch.org/docs/stable/nn.init.html
- Allow for flattening of x for greater shape conformity
- Allow for batching
- Add tests
- https://github.com/catid/audio_prediction/tree/master
- other classic RNN/LSTM tasks
References
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
torchxlstm-0.0.3.tar.gz
(4.3 kB
view details)
Built Distribution
File details
Details for the file torchxlstm-0.0.3.tar.gz.
File metadata
- Download URL: torchxlstm-0.0.3.tar.gz
- Upload date:
- Size: 4.3 kB
- Tags: Source
- Uploaded using Trusted Publishing? Yes
- Uploaded via: twine/5.0.0 CPython/3.12.3
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 23050af2559a0b20d8542bf6a6ab219c7aee1b13523b2e3c47a90ff8099725fe |
|
| MD5 | ee6f873f9ee630a0febbfe31979ebd16 |
|
| BLAKE2b-256 | 584028a31a2c02dbc4379043b675d13f901bb8f90c95ef710e5394534ac69b4e |
File details
Details for the file torchxlstm-0.0.3-py3-none-any.whl.
File metadata
- Download URL: torchxlstm-0.0.3-py3-none-any.whl
- Upload date:
- Size: 4.6 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? Yes
- Uploaded via: twine/5.0.0 CPython/3.12.3
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | 984abe69a05a5ca621aa0fc6dc8a286490de4fd7b917facca60b6e6bd1ab2139 |
|
| MD5 | a1aa454da85371798aa9b7eb4950ec80 |
|
| BLAKE2b-256 | 096137204fd3e7830676ff8fdb3b8edf172a4ae73a997da08c36fb5f0e7fe595 |
