jaxhps
Contents:
Spatial discretization
PDEProblem
API for solution methods
API for individual stages of HPS
Spectral discretization utilities
Examples
Data placement across accelerator devices
Contributing
jaxhps
Index
Index
A
|
B
|
C
|
D
|
E
|
F
|
G
|
I
|
J
|
L
|
M
|
N
|
P
|
Q
|
R
|
S
|
U
|
V
|
X
|
Y
|
Z
A
affine_transform() (in module jaxhps.quadrature)
B
barycentric_lagrange_interpolation_matrix_1D() (in module jaxhps.quadrature)
barycentric_lagrange_interpolation_matrix_2D() (in module jaxhps.quadrature)
barycentric_lagrange_interpolation_matrix_3D() (in module jaxhps.quadrature)
BD_inv_lst (jaxhps.PDEProblem attribute)
boundary_points (jaxhps.Domain attribute)
build_solver() (in module jaxhps)
C
chebyshev_points() (in module jaxhps.quadrature)
chebyshev_weights() (in module jaxhps.quadrature)
children (jaxhps.DiscretizationNode2D attribute)
(jaxhps.DiscretizationNode3D attribute)
D
D_inv_lst (jaxhps.PDEProblem attribute)
D_x_coefficients (jaxhps.PDEProblem attribute)
D_xx_coefficients (jaxhps.PDEProblem attribute)
D_xy_coefficients (jaxhps.PDEProblem attribute)
D_xz_coefficients (jaxhps.PDEProblem attribute)
D_y_coefficients (jaxhps.PDEProblem attribute)
D_yy_coefficients (jaxhps.PDEProblem attribute)
D_yz_coefficients (jaxhps.PDEProblem attribute)
D_z_coefficients (jaxhps.PDEProblem attribute)
D_zz_coefficients (jaxhps.PDEProblem attribute)
depth (jaxhps.DiscretizationNode2D attribute)
(jaxhps.DiscretizationNode3D attribute)
differentiation_matrix_1D() (in module jaxhps.quadrature)
DiscretizationNode2D (class in jaxhps)
DiscretizationNode3D (class in jaxhps)
Domain (class in jaxhps)
domain (jaxhps.PDEProblem attribute)
down_pass_adaptive_2D_DtN() (in module jaxhps.down_pass)
down_pass_adaptive_3D_DtN() (in module jaxhps.down_pass)
down_pass_uniform_2D_DtN() (in module jaxhps.down_pass)
down_pass_uniform_2D_ItI() (in module jaxhps.down_pass)
down_pass_uniform_3D_DtN() (in module jaxhps.down_pass)
downward_pass_subtree() (in module jaxhps)
E
eta (jaxhps.PDEProblem attribute)
F
from_adaptive_discretization() (jaxhps.Domain class method)
G
g_tilde_lst (jaxhps.PDEProblem attribute)
gauss_points() (in module jaxhps.quadrature)
get_adaptive_boundary_data_lst() (jaxhps.Domain method)
get_all_leaves() (in module jaxhps)
I
I_coefficients (jaxhps.PDEProblem attribute)
interior_points (jaxhps.Domain attribute)
interp_from_interior_points() (jaxhps.Domain method)
interp_to_interior_points() (jaxhps.Domain method)
J
jaxhps.down_pass
module
jaxhps.local_solve
module
jaxhps.merge
module
jaxhps.quadrature
module
jaxhps.up_pass
module
L
L (jaxhps.Domain attribute)
local_solve_stage_adaptive_2D_DtN() (in module jaxhps.local_solve)
local_solve_stage_adaptive_3D_DtN() (in module jaxhps.local_solve)
local_solve_stage_uniform_2D_DtN() (in module jaxhps.local_solve)
local_solve_stage_uniform_2D_ItI() (in module jaxhps.local_solve)
local_solve_stage_uniform_3D_DtN() (in module jaxhps.local_solve)
M
merge_stage_adaptive_2D_DtN() (in module jaxhps.merge)
merge_stage_adaptive_3D_DtN() (in module jaxhps.merge)
merge_stage_uniform_2D_DtN() (in module jaxhps.merge)
merge_stage_uniform_2D_ItI() (in module jaxhps.merge)
merge_stage_uniform_3D_DtN() (in module jaxhps.merge)
module
jaxhps.down_pass
jaxhps.local_solve
jaxhps.merge
jaxhps.quadrature
jaxhps.up_pass
N
n_0 (jaxhps.DiscretizationNode2D attribute)
(jaxhps.DiscretizationNode3D attribute)
n_1 (jaxhps.DiscretizationNode2D attribute)
(jaxhps.DiscretizationNode3D attribute)
n_2 (jaxhps.DiscretizationNode2D attribute)
(jaxhps.DiscretizationNode3D attribute)
n_3 (jaxhps.DiscretizationNode2D attribute)
(jaxhps.DiscretizationNode3D attribute)
n_4 (jaxhps.DiscretizationNode3D attribute)
n_5 (jaxhps.DiscretizationNode3D attribute)
n_leaves (jaxhps.Domain attribute)
nosource_local_solve_stage_uniform_2D_DtN() (in module jaxhps.local_solve)
nosource_local_solve_stage_uniform_2D_ItI() (in module jaxhps.local_solve)
nosource_merge_stage_uniform_2D_DtN() (in module jaxhps.merge)
nosource_merge_stage_uniform_2D_ItI() (in module jaxhps.merge)
P
p (jaxhps.Domain attribute)
PDEProblem (class in jaxhps)
Phi (jaxhps.PDEProblem attribute)
Q
q (jaxhps.Domain attribute)
R
reset() (jaxhps.PDEProblem method)
root (jaxhps.Domain attribute)
S
S_lst (jaxhps.PDEProblem attribute)
solve() (in module jaxhps)
solve_subtree() (in module jaxhps)
source (jaxhps.PDEProblem attribute)
U
up_pass_uniform_2D_DtN() (in module jaxhps.up_pass)
up_pass_uniform_2D_ItI() (in module jaxhps.up_pass)
update_coefficients() (jaxhps.PDEProblem method)
upward_pass_subtree() (in module jaxhps)
use_ItI (jaxhps.PDEProblem attribute)
V
v (jaxhps.PDEProblem attribute)
X
xmax (jaxhps.DiscretizationNode2D attribute)
(jaxhps.DiscretizationNode3D attribute)
xmin (jaxhps.DiscretizationNode2D attribute)
(jaxhps.DiscretizationNode3D attribute)
Y
Y (jaxhps.PDEProblem attribute)
ymax (jaxhps.DiscretizationNode2D attribute)
(jaxhps.DiscretizationNode3D attribute)
ymin (jaxhps.DiscretizationNode2D attribute)
(jaxhps.DiscretizationNode3D attribute)
Z
zmax (jaxhps.DiscretizationNode3D attribute)
zmin (jaxhps.DiscretizationNode3D attribute)