rwth-i6 / returnn

The RWTH extensible training framework for universal recurrent neural networks
http://returnn.readthedocs.io/
Other
350 stars 130 forks source link
deep-learning gpu recurrent-neural-networks tensorflow theano

================== Welcome to RETURNN

GitHub repository <https://github.com/rwth-i6/returnn>_. RETURNN paper 2016 <https://arxiv.org/abs/1608.00895>, RETURNN paper 2018 <https://arxiv.org/abs/1805.05225>_.

RETURNN - RWTH extensible training framework for universal recurrent neural networks, is a Theano/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:

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 <https://www.youtube.com/watch?v=wPKdYqSOlAY> (slides <https://www-i6.informatik.rwth-aachen.de/publications/download/1154/Zeyer--2020.pdf>) with an introduction of the core concepts.

More specific features include:

See documentation <https://returnn.readthedocs.io/>. See basic usage <https://returnn.readthedocs.io/en/latest/basic_usage.html>__ and technological overview <https://returnn.readthedocs.io/en/latest/tech_overview.html>.

Here is the video recording of a RETURNN overview talk <https://www-i6.informatik.rwth-aachen.de/web/Software/returnn/downloads/workshop-2019-01-29/01.recording.cut.mp4>_ (slides <https://www-i6.informatik.rwth-aachen.de/web/Software/returnn/downloads/workshop-2019-01-29/01.returnn-overview.session1.handout.v1.pdf>__, exercise sheet <https://www-i6.informatik.rwth-aachen.de/web/Software/returnn/downloads/workshop-2019-01-29/01.exercise_sheet.pdf>__; hosted by eBay).

There are many example demos <https://github.com/rwth-i6/returnn/blob/master/demos/>_ which work on artificially generated data, i.e. they should work as-is.

There are some real-world examples <https://github.com/rwth-i6/returnn-experiments>_ such as setups for speech recognition on the Switchboard or LibriSpeech corpus.

Some benchmark setups against other frameworks can be found here <https://github.com/rwth-i6/returnn-benchmarks>. The results are in the RETURNN paper 2016 <https://arxiv.org/abs/1608.00895>. Performance benchmarks of our LSTM kernel vs CuDNN and other TensorFlow kernels are in TensorFlow LSTM benchmark <https://returnn.readthedocs.io/en/latest/tf_lstm_benchmark.html>__.

There is also a wiki <https://github.com/rwth-i6/returnn/wiki>. Questions can also be asked on StackOverflow using the RETURNN tag <https://stackoverflow.com/questions/tagged/returnn>.

.. image:: https://github.com/rwth-i6/returnn/workflows/CI/badge.svg :target: https://github.com/rwth-i6/returnn/actions