derivkit.derivatives.autodiff package

Submodules

• derivkit.derivatives.autodiff.jax_autodiff module
  • Example:
  • Notes:
  • AutodiffDerivative
    • AutodiffDerivative.differentiate()
  • register_jax_autodiff_backend()
• derivkit.derivatives.autodiff.jax_core module
  • autodiff_derivative()
  • autodiff_gradient()
  • autodiff_hessian()
  • autodiff_jacobian()
• derivkit.derivatives.autodiff.jax_utils module
  • AutodiffUnavailable
  • apply_array_nd()
  • apply_scalar_1d()
  • apply_scalar_nd()
  • require_jax()
  • to_jax_array()
  • to_jax_scalar()

Module contents

JAX autodiff backend for DerivativeKit.