jTWA: Truncated Wigner on GPUs based on jax

Welcome to the documentation of jTWA. jTWA implements the Truncated Wigner Approximation in python using jax, thus allowing to write easily understandable python code that is readily executed on GPUs. For more information on jax see the documentation.

Getting Started

• Installation
• Quickstart

API Reference - General

• jTWA.integrate
• jTWA.util
• jTWA.visualization

API Reference - Spin 1

• jTWA.spin1