Exporting and serialization

Guides

• Exporting and serializing staged-out computations
  • Support for reverse-mode AD
  • Compatibility guarantees
  • Cross-platform and multi-platform export
  • Shape polymorphic export
  • Device-polymorphic export
  • Calling convention versions
  • Developer documentation
  • Migration guide from jax.experimental.export
• Shape polymorphism
  • Correctness of shape polymorphism
  • Computing with dimension variables
  • Debugging
• Interoperation with TensorFlow