graphax

• graphax package
  • Subpackages
    • graphax.examples package
      • Submodules
      • graphax.examples.deep_learning module
      • graphax.examples.differential_kinematics module
      • graphax.examples.easy module
      • graphax.examples.economics module
      • graphax.examples.minpack module
      • graphax.examples.neuromorphic module
      • graphax.examples.randoms module
      • graphax.examples.roe module
      • Module contents
    • graphax.sparse package
      • Submodules
      • graphax.sparse.tensor module
      • graphax.sparse.utils module
      • Module contents
  • Submodules
  • graphax.core module
    • append_pre_transforms()
    • jacve()
    • prepend_post_transforms()
    • tree_allclose()
    • unload_post_transforms()
    • unload_pre_transforms()
    • vertex_elimination_jaxpr()
  • graphax.equinox_bindings module
    • eqx_jacve()
    • filter_jacve()
  • graphax.primitives module
    • JacobianTransform
      • JacobianTransform.apply()
      • JacobianTransform.apply_inverse()
      • JacobianTransform.inverse_transform
      • JacobianTransform.transform
    • add_elemental_rule()
    • atan2_elemental_rule()
    • broadcast_elemental_rule()
    • concatenate_elemental_rule()
    • convert_element_type_rule()
    • defelemental()
    • defelemental2()
    • device_put_elemental_rule()
    • div_elemental_rule()
    • dot_general_elemental_rule()
    • eq_elemental_rule()
    • get_aval_shape()
    • get_ndim()
    • get_shape()
    • iota_elemental_rule()
    • make_parallel_jacobian()
    • max_elemental_rule()
    • min_elemental_rule()
    • mul_elemental_rule()
    • pjit_elemental_rule()
    • pow_elemental_rule()
    • reduce_elemental_rule()
    • reduce_max_elemental_rule()
    • reduce_min_elemental_rule()
    • reduce_sum_elemental_rule()
    • reshape_elemental_rule()
    • select_elemental_rule()
    • slice_elemental_rule()
    • squeeze_elemental_rule()
    • standard_elemental()
    • standard_elemental2()
    • stop_gradient_elemental_rule()
    • sub_elemental_rule()
    • transpose_elemental_rule()
    • with_type_promotion()
  • graphax.utils module
  • Module contents