Index A | B | C | D | E | F | G | J | 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) B box_mesh() (in module jax_fem.generate_mesh) box_mesh_gmsh() (in module jax_fem.generate_mesh) C cells (jax_fem.generate_mesh.Mesh attribute) check_mesh_TET4() (in module jax_fem.generate_mesh) compute_residual() (jax_fem.problem.Problem method) convert_from_dof_to_face_quad() (jax_fem.fe.FiniteElement method) convert_from_dof_to_quad() (jax_fem.fe.FiniteElement method) count_selected_faces() (jax_fem.generate_mesh.Mesh method) custom_init() (jax_fem.problem.Problem method) cylinder_mesh_gmsh() (in module jax_fem.generate_mesh) D dim (jax_fem.fe.FiniteElement attribute) (jax_fem.problem.Problem attribute) dirichlet_bc_info (jax_fem.fe.FiniteElement attribute) (jax_fem.problem.Problem attribute) Dirichlet_boundary_conditions() (jax_fem.fe.FiniteElement method) E ele_type (jax_fem.fe.FiniteElement attribute) (jax_fem.problem.Problem attribute) F FiniteElement (class in jax_fem.fe) G gauss_order (jax_fem.fe.FiniteElement attribute) (jax_fem.problem.Problem attribute) get_boundary_conditions_inds() (jax_fem.fe.FiniteElement method) get_elements() (in module jax_fem.basis) get_face_shape_grads() (jax_fem.fe.FiniteElement method) get_face_shape_vals_and_grads() (in module jax_fem.basis) get_meshio_cell_type() (in module jax_fem.generate_mesh) get_physical_quad_points() (jax_fem.fe.FiniteElement method) get_physical_surface_quad_points() (jax_fem.fe.FiniteElement method) get_shape_grads() (jax_fem.fe.FiniteElement method) get_shape_vals_and_grads() (in module jax_fem.basis) J jax_fem.basis module jax_fem.generate_mesh module jax_fem.utils module json_parse() (in module jax_fem.utils) L location_fns (jax_fem.problem.Problem attribute) M make_video() (in module jax_fem.utils) Mesh (class in jax_fem.generate_mesh) mesh (jax_fem.fe.FiniteElement attribute) (jax_fem.problem.Problem attribute) module jax_fem.basis jax_fem.generate_mesh jax_fem.utils N newton_update() (jax_fem.problem.Problem method) O optimize() (in module jax_fem.mma) P points (jax_fem.generate_mesh.Mesh attribute) 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) S save_sol() (in module jax_fem.utils) set_params() (jax_fem.problem.Problem method) sol_to_grad() (jax_fem.fe.FiniteElement method) solver() (in module jax_fem.solver) T timeit() (in module jax_fem.utils) U update_Dirichlet_boundary_conditions() (jax_fem.fe.FiniteElement method) V vec (jax_fem.fe.FiniteElement attribute) (jax_fem.problem.Problem attribute) W walltime() (in module jax_fem.utils)