Public API

  • API Documentation
adv-jax-math
  • Index

Index

B | S

B

  • batch_jacfwd() (in module adv_jax_math)
  • batch_jacrev() (in module adv_jax_math)
  • batch_map() (in module adv_jax_math)
  • batch_vectorize() (in module adv_jax_math)
  • batch_vmap() (in module adv_jax_math)

S

  • sparse_pullback() (in module adv_jax_math)
  • sparse_pullback_map() (in module adv_jax_math)

© Copyright 2026, Kaya Unalmis. Last updated on Jul 26, 2026.

Built with Sphinx using a theme provided by Read the Docs.