class_docstr: Jax pytree representing a lazy dot operation.
class_docstr_tok: ["Jax","pytree","representing","a","lazy","dot","operation","."]
docstr: Function used by ``jax`` to flatten the JaxLazyDot operation.
  
  Returns:
      tuple: tuple containing children and the auxiliary data of the class
docstr_tok: ["Function","used","by","``","jax","``","to","flatten","the","JaxLazyDot","operation",".","Returns",":","tuple",":","tuple","containing","children","and","the","auxiliary","data","of","the","class"]
code: def tree_flatten(self):
      """Function used by ``jax`` to flatten the JaxLazyDot operation.
  
      Returns:
          tuple: tuple containing children and the auxiliary data of the class
      """
      children = (self.coeffs, self.mats)
      aux_data = None
      return (children, aux_data)
  
file_path: pennylane/pennylane/pulse/parametrized_hamiltonian_pytree.py
imports: from dataclasses import dataclass
  from typing import Callable, Optional, Tuple, Union
  import jax
  import jax.numpy
  from jax.experimental import sparse
  from jax.tree_util import register_pytree_node_class
  import pennylane
  from .parametrized_hamiltonian import ParametrizedHamiltonian
  from .hardware_hamiltonian import HardwareHamiltonian
label_desc: Use image node_id 3 for calling the LazyDotPytree obj's underlying member method code with example usage: obj.tree_flatten() and returns: children,  aux_data