Index A | B | C | D | E | F | G | I | J | K | L | M | N | O | P | R | S | T | U | V | W A ad_wrapper() (in module jax_fem.solver) additional_info (jax_fem.problem.Problem attribute) apply_bc() (in module jax_fem.solver) apply_bc_vec() (in module jax_fem.solver) applyDensityFilter() (in module jax_fem.mma) applySensitivityFilter() (in module jax_fem.mma) arc_length_solver_disp_driven() (in module jax_fem.solver) arc_length_solver_force_driven() (in module jax_fem.solver) args (jax_fem.mma.partial attribute) assembleCSR() (in module jax_fem.solver) assign_bc() (in module jax_fem.solver) assign_ones_bc() (in module jax_fem.solver) assign_zeros_bc() (in module jax_fem.solver) B box_mesh() (in module jax_fem.generate_mesh) box_mesh_gmsh() (in module jax_fem.generate_mesh), [1] C calC() (in module jax_fem.solver) check_mesh_TET4() (in module jax_fem.generate_mesh) compute_face() (jax_fem.problem.Problem method) compute_filter_kd_tree() (in module jax_fem.mma) compute_newton_vars() (jax_fem.problem.Problem method) compute_residual() (jax_fem.problem.Problem method) compute_residual_vars() (jax_fem.problem.Problem method) compute_residual_vars_helper() (jax_fem.problem.Problem method) convert_from_dof_to_face_quad() (jax_fem.fe.FiniteElement method) (jax_fem.problem.FiniteElement method) convert_from_dof_to_quad() (jax_fem.fe.FiniteElement method) (jax_fem.problem.FiniteElement method) copy_bc() (in module jax_fem.solver) count_selected_faces() (jax_fem.fe.Mesh method) (jax_fem.generate_mesh.Mesh method) (jax_fem.problem.Mesh method) custom_init() (jax_fem.problem.Problem method) cylinder_mesh_gmsh() (in module jax_fem.generate_mesh) D dataclass() (in module jax_fem.fe) (in module jax_fem.problem) dim (jax_fem.fe.FiniteElement attribute), [1] (jax_fem.problem.FiniteElement attribute), [1] (jax_fem.problem.Problem attribute) dirichlet_bc_info (jax_fem.fe.FiniteElement attribute), [1] (jax_fem.problem.FiniteElement attribute), [1] (jax_fem.problem.Problem attribute) Dirichlet_boundary_conditions() (jax_fem.fe.FiniteElement method) (jax_fem.problem.FiniteElement method) dynamic_relax_solve() (in module jax_fem.solver), [1] E ele_type (jax_fem.fe.FiniteElement attribute), [1] (jax_fem.problem.FiniteElement attribute), [1] (jax_fem.problem.Problem attribute) F FiniteElement (class in jax_fem.fe) (class in jax_fem.problem) func (jax_fem.mma.partial attribute) G gauss_order (jax_fem.fe.FiniteElement attribute) (jax_fem.problem.FiniteElement attribute) (jax_fem.problem.Problem attribute) get_A() (in module jax_fem.solver) get_boundary_conditions_inds() (jax_fem.fe.FiniteElement method) (jax_fem.problem.FiniteElement method) get_elements() (in module jax_fem.basis) (in module jax_fem.generate_mesh) get_face_shape_grads() (jax_fem.fe.FiniteElement method) (jax_fem.problem.FiniteElement method) get_face_shape_vals_and_grads() (in module jax_fem.basis) (in module jax_fem.fe) (in module jax_fem.generate_mesh) get_flatten_fn() (in module jax_fem.solver) get_laplace_kernel() (jax_fem.problem.Problem method) get_mass_kernel() (jax_fem.problem.Problem method) get_meshio_cell_type() (in module jax_fem.generate_mesh) (in module jax_fem.utils) get_physical_quad_points() (jax_fem.fe.FiniteElement method) (jax_fem.problem.FiniteElement method) get_physical_surface_quad_points() (jax_fem.fe.FiniteElement method) (jax_fem.problem.FiniteElement method) get_q_vec() (in module jax_fem.solver) get_shape_grads() (jax_fem.fe.FiniteElement method) (jax_fem.problem.FiniteElement method) get_shape_vals_and_grads() (in module jax_fem.basis) (in module jax_fem.fe) get_surface_kernel() (jax_fem.problem.Problem method) getAsymptoteValues() (jax_fem.mma.MMA method) getLagrangeMultipliers() (jax_fem.mma.MMA method) getOptimalValues() (jax_fem.mma.MMA method) getSlackValue() (jax_fem.mma.MMA method) I implicit_vjp() (in module jax_fem.solver) J jax_fem.basis module jax_fem.fe module jax_fem.generate_mesh module jax_fem.logger_setup module jax_fem.mma module jax_fem.problem module jax_fem.solver module jax_fem.utils module jax_solve() (in module jax_fem.solver) json_parse() (in module jax_fem.utils) K keywords (jax_fem.mma.partial attribute) L line_search() (in module jax_fem.solver) linear_incremental_solver() (in module jax_fem.solver) linear_solver() (in module jax_fem.solver) location_fns (jax_fem.problem.Problem attribute) M make_video() (in module jax_fem.utils) Mesh (class in jax_fem.fe) (class in jax_fem.generate_mesh) (class in jax_fem.problem) mesh (jax_fem.fe.FiniteElement attribute), [1] (jax_fem.problem.FiniteElement attribute), [1] (jax_fem.problem.Problem attribute) MMA (class in jax_fem.mma) mmasub() (jax_fem.mma.MMA method) modify_vtu_file() (in module jax_fem.utils) module jax_fem.basis jax_fem.fe jax_fem.generate_mesh jax_fem.logger_setup jax_fem.mma jax_fem.problem jax_fem.solver jax_fem.utils N newton_update() (jax_fem.problem.Problem method) O operator_to_matrix() (in module jax_fem.solver) optimize() (in module jax_fem.mma) P partial (class in jax_fem.mma) periodic_bc_info (jax_fem.fe.FiniteElement attribute), [1] (jax_fem.problem.FiniteElement attribute), [1] petsc_solve() (in module jax_fem.solver) pre_jit_fns() (jax_fem.problem.Problem method) print_BC_info() (jax_fem.fe.FiniteElement method) (jax_fem.problem.FiniteElement method) printInfo() (in module jax_fem.solver) Problem (class in jax_fem.problem) R read_abaqus_and_write_vtk() (in module jax_fem.utils) rectangle_mesh() (in module jax_fem.generate_mesh) registerMMAIter() (jax_fem.mma.MMA method) reorder_inds() (in module jax_fem.basis) resetMMACounter() (jax_fem.mma.MMA method) S save_sol() (in module jax_fem.utils) set_params() (jax_fem.problem.Problem method) setConstraintWithGradient() (jax_fem.mma.MMA method) setLowerAndUpperAsymptotes() (jax_fem.mma.MMA method) setMinandMaxBoundsForDesignVariables() (jax_fem.mma.MMA method) setMoveLimit() (jax_fem.mma.MMA method) setNumConstraints() (jax_fem.mma.MMA method) setNumDesignVariables() (jax_fem.mma.MMA method) setObjectiveWithGradient() (jax_fem.mma.MMA method) setScalingParams() (jax_fem.mma.MMA method) setup_logger() (in module jax_fem.logger_setup) sol_to_grad() (jax_fem.fe.FiniteElement method) (jax_fem.problem.FiniteElement method) solver() (in module jax_fem.solver) split_and_compute_cell() (jax_fem.problem.Problem method) subsolv() (in module jax_fem.mma) T timeit() (in module jax_fem.problem) (in module jax_fem.utils) U umfpack_solve() (in module jax_fem.solver) update_Dirichlet_boundary_conditions() (jax_fem.fe.FiniteElement method) (jax_fem.problem.FiniteElement method) V vec (jax_fem.fe.FiniteElement attribute), [1] (jax_fem.problem.FiniteElement attribute), [1] (jax_fem.problem.Problem attribute) W walltime() (in module jax_fem.utils) wraps() (in module jax_fem.utils)