Skip to content

Backend framework comparison

Jump to bottom
Albert Zeyer edited this page Nov 8, 2021 · 9 revisions

We want to compare TensorFlow, PyTorch, JAX and maybe other similar frameworks here.

We do not want to compare higher-level frameworks like Keras here.

Side node: The classification low-level, mid-level and high-level framework comes from our tutorial (video, slides).

This comparison is specifically about their properties specifically when used as a backend framework for RETURNN, using the RETURNN principles and core features.

Clone this wiki locally