{"ast_node_type":"other","tokens":[{"end":23,"kind":"keyword","start":0,"text":"pub"}]}
    pub mode: DiffMode,
    /// Documentation](https://jax.readthedocs.io/en/latest/_tutorials/advanced-autodiff.html#how-it-s-made-two-foundational-autodiff-functions).
    /// A user-provided, batching width. If not given, we will default to 1 (no batching).