Index

A

add_error (physika.utils.infer_expr.ExprContext attribute) animate() (in module physika.runtime) ast_to_torch_expr() (in module physika.utils.ast_utils)

ast_uses_func() (in module physika.utils.ast_utils) ast_uses_solve() (in module physika.utils.ast_utils) ast_uses_sympy() (in module physika.utils.ast_utils)

B

bernoulli_dist() (in module physika.features.randomness) beta_dist() (in module physika.features.randomness)

broadcast_op() (in module physika.utils.type_checker_utils) build_class() (in module physika.features.classes) build_unified_ast() (in module physika.utils.ast_utils)

C

class_env (physika.utils.infer_expr.ExprContext attribute) ClassFeature (class in physika.features.classes)

collect_grad_targets() (in module physika.utils.ast_utils) compute_grad() (in module physika.runtime) condition_to_expr() (in module physika.utils.ast_utils)

D

dims_compatible() (in module physika.utils.type_checker_utils)

E

emit_body_stmts() (in module physika.utils.ast_utils) emit_for_stmts() (in module physika.utils.ast_utils) emit_func_loop_body() (in module physika.utils.ast_utils) emit_method() (in module physika.features.classes) env (physika.utils.infer_expr.ExprContext attribute) evaluate() (in module physika.runtime) expr_add_sub() (in module physika.utils.infer_expr) expr_array() (in module physika.utils.infer_expr) expr_call() (in module physika.utils.infer_expr) expr_chain_index() (in module physika.utils.infer_expr) expr_complex() (in module physika.utils.infer_expr) expr_cond() (in module physika.utils.infer_expr) expr_div() (in module physika.utils.infer_expr)

expr_for_expr() (in module physika.utils.infer_expr) expr_for_expr_range() (in module physika.utils.infer_expr) expr_imaginary() (in module physika.utils.infer_expr) expr_index() (in module physika.utils.infer_expr) expr_indexN() (in module physika.utils.infer_expr) expr_matmul() (in module physika.utils.infer_expr) expr_mul() (in module physika.utils.infer_expr) expr_neg() (in module physika.utils.infer_expr) expr_num() (in module physika.utils.infer_expr) expr_pow() (in module physika.utils.infer_expr) expr_slice() (in module physika.utils.infer_expr) expr_var() (in module physika.utils.infer_expr) ExprContext (class in physika.utils.infer_expr) extract_dist_args() (in module physika.features.randomness)

F

find_indexed_arrays() (in module physika.utils.parser_utils) forward_rules() (physika.features.classes.ClassFeature method) (physika.features.randomness.RandomnessFeature method)

from_ast_to_torch() (in module physika.codegen) from_typespec() (in module physika.utils.type_checker_utils) func_env (physika.utils.infer_expr.ExprContext attribute)

G

gamma_dist() (in module physika.features.randomness) generate_class() (in module physika.features.classes) (in module physika.utils.ast_utils) generate_function() (in module physika.utils.ast_utils)

generate_statement() (in module physika.utils.ast_utils) get_dim() (in module physika.features.randomness) get_line_info() (in module physika.utils.type_checker_utils) get_shape_args() (in module physika.features.randomness) get_tensor_shape() (in module physika.utils.type_checker_utils)

I

infer_expr() (in module physika.utils.infer_expr)

is_learnable() (in module physika.features.classes)

L

lexer_rules() (physika.features.classes.ClassFeature method) (physika.features.randomness.RandomnessFeature method)

M

make_parser_rules() (in module physika.features.classes) make_tensor() (in module physika.utils.type_checker_utils) make_tensor_type() (in module physika.utils.type_checker_utils) matmul_op() (in module physika.utils.type_checker_utils) module physika.codegen physika.features.classes physika.features.randomness physika.parser physika.runtime physika.type_checker physika.utils.ast_utils physika.utils.infer_expr physika.utils.parser_utils physika.utils.print_utils physika.utils.type_checker_utils

N

name (physika.features.classes.ClassFeature attribute) (physika.features.randomness.RandomnessFeature attribute)

normal_dist() (in module physika.features.randomness)

O

occurs_in() (in module physika.utils.type_checker_utils)

P

p_args_empty() (in module physika.parser) p_args_multi() (in module physika.parser) p_args_single() (in module physika.parser) p_condition_eq() (in module physika.parser) p_condition_geq() (in module physika.parser) p_condition_gt() (in module physika.parser) p_condition_leq() (in module physika.parser) p_condition_lt() (in module physika.parser) p_condition_neq() (in module physika.parser) p_dimension_contravariant() (in module physika.parser) p_dimension_covariant() (in module physika.parser) p_dimension_invariant() (in module physika.parser) p_dimension_invariant_id() (in module physika.parser) p_dimension_list_multi() (in module physika.parser) p_dimension_list_single() (in module physika.parser) p_dimension_type_as_symbol() (in module physika.parser) p_elements_multi() (in module physika.parser) p_elements_newline() (in module physika.parser) p_elements_single() (in module physika.parser) p_error() (in module physika.parser) p_expr_minus() (in module physika.parser) p_expr_plus() (in module physika.parser) p_expr_term() (in module physika.parser) p_factor_array() (in module physika.parser) p_factor_call() (in module physika.parser) p_factor_complex() (in module physika.parser) p_factor_for_expr() (in module physika.parser) p_factor_for_expr_auto() (in module physika.parser) p_factor_for_expr_range() (in module physika.parser) p_factor_group() (in module physika.parser) p_factor_id() (in module physika.parser) p_factor_index() (in module physika.parser) p_factor_index_var() (in module physika.parser) p_factor_indexN() (in module physika.parser) p_factor_neg() (in module physika.parser) p_factor_number() (in module physika.parser) p_factor_slice() (in module physika.parser) p_factor_string() (in module physika.parser) p_for_body_empty() (in module physika.parser) p_for_body_multi() (in module physika.parser) p_for_statement_assign() (in module physika.parser) p_for_statement_call() (in module physika.parser) p_for_statement_empty() (in module physika.parser) p_for_statement_for() (in module physika.parser) p_for_statement_for_range() (in module physika.parser) p_for_statement_if_else() (in module physika.parser) p_for_statement_if_only() (in module physika.parser) p_for_statement_index_assign_nd() (in module physika.parser) p_for_statement_pluseq() (in module physika.parser) p_func_args_empty() (in module physika.parser) p_func_args_multi() (in module physika.parser) p_func_args_single() (in module physika.parser) p_func_body_stmt_assign() (in module physika.parser) p_func_body_stmt_decl() (in module physika.parser) p_func_body_stmt_empty() (in module physika.parser) p_func_body_stmt_for() (in module physika.parser) p_func_body_stmt_for_implicit() (in module physika.parser) p_func_body_stmt_for_range() (in module physika.parser) p_func_body_stmt_if_else() (in module physika.parser) p_func_body_stmt_if_else_return() (in module physika.parser) p_func_body_stmt_if_only() (in module physika.parser) p_func_body_stmt_if_return() (in module physika.parser) p_func_body_stmt_index_assign() (in module physika.parser) p_func_body_stmt_index_assign_nd() (in module physika.parser) p_func_body_stmt_tuple_unpack() (in module physika.parser) p_func_body_stmt_tuple_unpack_three() (in module physika.parser) p_func_body_stmt_zeros_decl() (in module physika.parser) p_func_body_stmts_multi() (in module physika.parser) p_func_body_stmts_single() (in module physika.parser) p_func_elements_multi() (in module physika.parser) p_func_elements_single() (in module physika.parser) p_func_expr_minus() (in module physika.parser) p_func_expr_plus() (in module physika.parser) p_func_expr_term() (in module physika.parser) p_func_factor_array() (in module physika.parser) p_func_factor_call() (in module physika.parser) p_func_factor_call_index() (in module physika.parser) p_func_factor_chain_index() (in module physika.parser) p_func_factor_complex() (in module physika.parser) p_func_factor_for_expr() (in module physika.parser) p_func_factor_for_expr_auto() (in module physika.parser) p_func_factor_for_expr_range() (in module physika.parser) p_func_factor_group() (in module physika.parser) p_func_factor_id() (in module physika.parser) p_func_factor_imaginary() (in module physika.parser) p_func_factor_index() (in module physika.parser) p_func_factor_indexN() (in module physika.parser) p_func_factor_number() (in module physika.parser)

p_func_factor_step_slice() (in module physika.parser) p_func_factor_string() (in module physika.parser) p_func_init_empty() (in module physika.parser) p_func_init_multi() (in module physika.parser) p_func_init_stmt_assign() (in module physika.parser) p_func_init_stmt_empty() (in module physika.parser) p_func_loop_body_empty() (in module physika.parser) p_func_loop_body_multi() (in module physika.parser) p_func_loop_stmt_assign() (in module physika.parser) p_func_loop_stmt_empty() (in module physika.parser) p_func_loop_stmt_for_range() (in module physika.parser) p_func_loop_stmt_if() (in module physika.parser) p_func_loop_stmt_if_else() (in module physika.parser) p_func_loop_stmt_index_assign_nd() (in module physika.parser) p_func_loop_stmt_index_pluseq() (in module physika.parser) p_func_loop_stmt_pluseq() (in module physika.parser) p_func_power_factor() (in module physika.parser) p_func_power_neg() (in module physika.parser) p_func_power_pow() (in module physika.parser) p_func_term_power() (in module physika.parser) p_func_term_times() (in module physika.parser) p_id_list() (in module physika.parser) p_import_list_multiple() (in module physika.parser) p_import_list_single() (in module physika.parser) p_loop_index_list_multi() (in module physika.parser) p_loop_index_list_single() (in module physika.parser) p_loop_var_list_multi() (in module physika.parser) p_loop_var_list_single() (in module physika.parser) p_multi_index_list_base() (in module physika.parser) p_multi_index_list_extend() (in module physika.parser) p_params_empty() (in module physika.parser) p_params_multi() (in module physika.parser) p_params_single() (in module physika.parser) p_program() (in module physika.parser) p_statement_assign() (in module physika.parser) p_statement_decl() (in module physika.parser) p_statement_empty() (in module physika.parser) p_statement_expr() (in module physika.parser) p_statement_for() (in module physika.parser) p_statement_for_range() (in module physika.parser) p_statement_function() (in module physika.parser) p_statement_function_body_only() (in module physika.parser) p_statement_function_decl() (in module physika.parser) p_statement_function_multi_decl() (in module physika.parser) p_statement_function_with_body() (in module physika.parser) p_statement_function_with_loop() (in module physika.parser) p_statement_if_else() (in module physika.parser) p_statement_if_only() (in module physika.parser) p_statement_import() (in module physika.parser) p_statement_index_assign() (in module physika.parser) p_statement_index_assign_nd() (in module physika.parser) p_statement_symbol_decl() (in module physika.parser) p_statement_symbol_multi_decl() (in module physika.parser) p_statements_multi() (in module physika.parser) p_statements_single() (in module physika.parser) p_statemet_equation_decl() (in module physika.parser) p_term_binop() (in module physika.parser) p_term_factor() (in module physika.parser) p_type_function() (in module physika.parser) p_type_scalar() (in module physika.parser) p_type_tangent() (in module physika.parser) p_type_tensor() (in module physika.parser) parser_rules() (physika.features.classes.ClassFeature method) (physika.features.randomness.RandomnessFeature method) physika.codegen module physika.features.classes module physika.features.randomness module physika.parser module physika.runtime module physika.type_checker module physika.utils.ast_utils module physika.utils.infer_expr module physika.utils.parser_utils module physika.utils.print_utils module physika.utils.type_checker_utils module physika_print() (in module physika.runtime) print_type_check_results() (in module physika.utils.print_utils) print_unified_ast() (in module physika.utils.print_utils)

R

RandomnessFeature (class in physika.features.randomness) replace_class_params() (in module physika.features.classes) (in module physika.utils.ast_utils)

run() (physika.type_checker.TypeChecker method)

S

s (physika.utils.infer_expr.ExprContext attribute) sample() (in module physika.features.randomness) shapes_broadcast_compatible() (in module physika.utils.type_checker_utils)

simulate() (in module physika.runtime) solve() (in module physika.runtime) statement_check() (in module physika.utils.type_checker_utils)

T

train() (in module physika.runtime) type_infer() (in module physika.utils.type_checker_utils) type_promotion() (in module physika.utils.type_checker_utils) type_rules() (physika.features.classes.ClassFeature method) (physika.features.randomness.RandomnessFeature method)

type_to_str() (in module physika.utils.type_checker_utils) TypeChecker (class in physika.type_checker) types_compatible() (in module physika.utils.type_checker_utils)

U

uniform_dist() (in module physika.features.randomness) unify() (in module physika.utils.type_checker_utils)

unify_dim() (in module physika.utils.type_checker_utils) unwrap_return() (in module physika.features.classes)