qutip-jax : Jax-backend for QuTiP

Introduction and installation

• Introduction
  • The qutip-jax package
• Installation
  • Prerequisites

User guide

• Jax in QuTiP
  • Basic usage
• Using JAX with QuTiP Core Metrics and Entropy
  • Setting Up the JAX Backend
  • Using jax.jit with QuTiP
  • Using jax.grad with QuTiP
  • Changing Back to Default Backend
• Jax Backend in QuTiP’s Solver
  • Using Jax in mesolve
  • Using Jax in mcsolve
• Auto differentiation in Qutip’s Solver
  • Auto differentiation in sesolve
  • Auto differentiation in mcsolve

API documentation

• qutip_jax package
• Diffrax ODE
  • DiffraxIntegrator

Changelog

• Changelog
  • Version 0.0.0.dev

Bibliography

• Bibliography