Index A | B | C | D | E | F | G | I | L | M | P | R | S | T | U | W | Z A abstract_rules (in module autoform.extend) active_client (in module autoform.extend) active_interpreter (in module autoform.extend) active_tags (in module autoform.extend) autoform.__version__ (built-in variable) autoform.PYTREE_NAMESPACE (built-in variable) AVal (class in autoform.extend) avalof() (in module autoform.extend) B batch() (in module autoform) batch_call_p (in module autoform.extend) batch_index() (in module autoform.extend) batch_rules (in module autoform.extend) batch_spec() (in module autoform.extend) batch_transpose() (in module autoform.extend) Bool (class in autoform) BoolAVal (class in autoform.extend) C checkpoint() (in module autoform) checkpoint_p (in module autoform.extend) collect() (in module autoform) concat() (in module autoform) concat_p (in module autoform.extend) custom() (in module autoform) D dce() (in module autoform) depends() (in module autoform) depends_p (in module autoform.extend) Doc (class in autoform) E Enum (class in autoform) F factor() (in module autoform) Float (class in autoform) FloatAVal (class in autoform.extend) fold() (in module autoform) format() (in module autoform) format_p (in module autoform.extend) G gather_p (in module autoform.extend) I impl_rules (in module autoform.extend) inject() (in module autoform) Int (class in autoform) IntAVal (class in autoform.extend) Interpreter (class in autoform.extend) IR (class in autoform.extend) ir_aval() (in module autoform.extend) IREqn (class in autoform.extend) IRVar (class in autoform.extend) is_irvar() (in module autoform.extend) is_zero() (in module autoform.extend) L lm_call() (in module autoform) lm_call_p (in module autoform.extend) lm_client() (in module autoform) lm_schema_call() (in module autoform) lm_schema_call_p (in module autoform.extend) M match() (in module autoform) match_p (in module autoform.extend) materialize() (in module autoform.extend) memoize() (in module autoform) P Prim (class in autoform.extend) pull_bwd_rules (in module autoform.extend) pull_fwd_rules (in module autoform.extend) pullback() (in module autoform) pullback_call_p (in module autoform.extend) push_rules (in module autoform.extend) pushforward() (in module autoform) pushforward_call_p (in module autoform.extend) R register_add() (in module autoform.extend) register_cotangent_accumulator() (in module autoform.extend) register_div() (in module autoform.extend) register_eq() (in module autoform.extend) register_matmul() (in module autoform.extend) register_mul() (in module autoform.extend) register_non_dce() (in module autoform.extend) register_non_memoizable() (in module autoform.extend) register_sub() (in module autoform.extend) register_trace_type() (in module autoform.extend) register_zero() (in module autoform.extend) S sched() (in module autoform) serial_fanout() (in module autoform.extend) stop_gradient() (in module autoform) stop_gradient_p (in module autoform.extend) Str (class in autoform) StrAVal (class in autoform.extend) switch() (in module autoform) switch_p (in module autoform.extend) T tag() (in module autoform) trace() (in module autoform) U using_interpreter() (in module autoform.extend) W weighted() (in module autoform) while_loop() (in module autoform) while_loop_p (in module autoform.extend) Z Zero (class in autoform.extend) zeroof() (in module autoform.extend)