Welcome to RETURNN
GitHub repository. RETURNN paper 2016, RETURNN paper 2018.
RETURNN - RWTH extensible training framework for universal recurrent neural networks, is a PyTorch/TensorFlow-based implementation of modern recurrent neural network architectures. It is optimized for fast and reliable training of recurrent neural networks in a multi-GPU environment.
The high-level features and goals of RETURNN are:
Simplicity
Writing config / code is simple & straight-forward (setting up experiment, defining model)
Debugging in case of problems is simple
Reading config / code is simple (defined model, training, decoding all becomes clear)
Flexibility
Allow for many different kinds of experiments / models
Efficiency
Training speed
Decoding speed
All items are important for research, decoding speed is esp. important for production.
See our Interspeech 2020 tutorial “Efficient and Flexible Implementation of Machine Learning for ASR and MT” video (slides) with an introduction of the core concepts.
More specific features include:
Mini-batch training of feed-forward neural networks
Sequence-chunking based batch training for recurrent neural networks
Long short-term memory recurrent neural networks including our own fast CUDA kernel
Multidimensional LSTM (GPU only, there is no CPU version)
Memory management for large data sets
Work distribution across multiple devices
Flexible and fast architecture which allows all kinds of encoder-attention-decoder models
See documentation. See basic usage and technological overview.
Here is the video recording of a RETURNN overview talk (slides, exercise sheet; hosted by eBay).
There are many example demos which work on artificially generated data, i.e. they should work as-is.
There are some real-world examples such as setups for speech recognition on the Switchboard or LibriSpeech corpus.
Some benchmark setups against other frameworks can be found here. The results are in the RETURNN paper 2016. Performance benchmarks of our LSTM kernel vs CuDNN and other TensorFlow kernels are in TensorFlow LSTM benchmark.
There is also a wiki. Questions can also be asked on StackOverflow using the RETURNN tag.
Dependencies
pip dependencies are listed in requirements.txt and requirements-dev, although some parts of the code may require additional dependencies (e.g. librosa, resampy) on-demand.
RETURNN supports Python >= 3.8. Bumps to the minimum Python version are listed in CHANGELOG.md.
TensorFlow-based setups require TensorFlow >= 2.2.
PyTorch-based setups require Torch >= 1.0.
1 maintainer
Metadata
1 maintainer
Release files for returnn 1.20260824.190105
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | Provenance | Actions |
|---|---|---|---|---|
| returnn-1.20260824.190105.tar.gz | 2.8 MB | Without provenance | Details for returnn-1.20260824.190105.tar.gz |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Actions Reset |
|---|---|---|---|---|
| returnn-1.20260824.190105-py3-none-any.whl | Python 3 | none | any | Details for returnn-1.20260824.190105-py3-none-any.whl |
Total release size: 4.6 MB
Release files / returnn-1.20260824.190105.tar.gz
| Download URL | returnn-1.20260824.190105.tar.gz |
|---|---|
| Size | 2.8 MB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
c1223356b03b592b15f246b0030a7f75dd35b8a1b93f7904b390c07283746507
|
|
BLAKE2b-256 checksum How to use checksums |
295da0f52683505ec7c10c7e8b084d847837658609ecbc254788ccf1f58ea6ca
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/6.2.0 CPython/3.9.25
|
Release files / returnn-1.20260824.190105-py3-none-any.whl
| Download URL | returnn-1.20260824.190105-py3-none-any.whl |
|---|---|
| Size | 1.8 MB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
903e030ed5b08c6696adf6f78e342b484c4cdb942d07be822857b6d51f3eec83
|
|
BLAKE2b-256 checksum How to use checksums |
d02ee4c665b6d080317e6d2c6984eee8c9bd3e7a4fc976f98d66c944c30bbb2c
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/6.2.0 CPython/3.9.25
|
Security for returnn 1.20260824.190105
Provenance
Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.
No provenance found
None of the files in this release were uploaded with provenance metadata.
A note on package security
PyPI is a repository for distributing packages, not a service for auditing them. While third-party security researchers and our staff do their best to detect and remove malicious packages, PyPI cannot guarantee the quality, security or suitability of any given package.
To learn more about protecting yourself from malicious actors, visit our package security documentation.
