LICENSE
README.md
pyproject.toml
jax_numerical_operators.egg-info/PKG-INFO
jax_numerical_operators.egg-info/SOURCES.txt
jax_numerical_operators.egg-info/dependency_links.txt
jax_numerical_operators.egg-info/requires.txt
jax_numerical_operators.egg-info/top_level.txt
jno/__init__.py
jno/_fem.py
jno/bayesian.py
jno/core.py
jno/dd.py
jno/differential_operators.py
jno/fdm.py
jno/fn.py
jno/info.py
jno/integration_operators.py
jno/jnp_ops.py
jno/litho.py
jno/lora.py
jno/noise.py
jno/precond.py
jno/py.typed
jno/rcwa.py
jno/solve.py
jno/stencils.py
jno/trace_compiler.py
jno/trace_evaluator.py
jno/trackers.py
jno/tuner.py
jno/architectures/__init__.py
jno/architectures/common.py
jno/architectures/linear.py
jno/architectures/lora.py
jno/architectures/models.py
jno/architectures/pdeformer2_bridge.py
jno/domain/__init__.py
jno/domain/boundary_region.py
jno/domain/domain_class.py
jno/domain/domain_data.py
jno/domain/enclosure.py
jno/domain/geometries.py
jno/domain/mesh_utils.py
jno/domain/meshio_mixin.py
jno/domain/polygon_domain.py
jno/domain/simplex_pool.py
jno/domain/view_factor_closed.py
jno/geometry/__init__.py
jno/geometry/emit.py
jno/geometry/naming.py
jno/geometry/path.py
jno/geometry/primitives.py
jno/geometry/shape.py
jno/optimizers/__init__.py
jno/optimizers/continuation.py
jno/optimizers/engd.py
jno/optimizers/md_decouple.py
jno/optimizers/mma.py
jno/optimizers/soap.py
jno/optimizers/ssbroyden.py
jno/trace/__init__.py
jno/trace/unit_log.py
jno/trace/units.py
jno/trace/views.py
jno/utils/__init__.py
jno/utils/ad_mode.py
jno/utils/config.py
jno/utils/dtypes.py
jno/utils/explainability.py
jno/utils/iree.py
jno/utils/load_save.py
jno/utils/logger.py
jno/utils/optics.py
jno/utils/profiling.py
jno/utils/schemes.py
jno/utils/statistics.py
jno/utils/adaptive/__init__.py
jno/utils/adaptive/callbacks.py
jno/utils/adaptive/lrscheduler.py
jno/utils/adaptive/resampling.py
jno/utils/adaptive/weights.py
jno/utils/solver/__init__.py
jno/utils/solver/amg.py
jno/utils/solver/ams.py
jno/utils/solver/arclength.py
jno/utils/solver/backend_blocks.py
jno/utils/solver/contact_search.py
jno/utils/solver/eigen.py
jno/utils/solver/element_operator.py
jno/utils/solver/fem_1d.py
jno/utils/solver/fem_adapt.py
jno/utils/solver/fem_cover.py
jno/utils/solver/fem_elements.py
jno/utils/solver/fem_facets.py
jno/utils/solver/fem_lagrange.py
jno/utils/solver/fem_native.py
jno/utils/solver/fem_nonnodal.py
jno/utils/solver/fem_refine.py
jno/utils/solver/fem_route.py
jno/utils/solver/fem_topology.py
jno/utils/solver/fem_utils.py
jno/utils/solver/fsai.py
jno/utils/solver/history_march.py
jno/utils/solver/krylov.py
jno/utils/solver/lattice.py
jno/utils/solver/lattice_mg.py
jno/utils/solver/linear.py
jno/utils/solver/march_checkpoint.py
jno/utils/solver/mass.py
jno/utils/solver/matfun.py
jno/utils/solver/matvec_format.py
jno/utils/solver/newton_krylov.py
jno/utils/solver/parametric_helpers.py
jno/utils/solver/placement.py
jno/utils/solver/reconnect.py
jno/utils/solver/schwarz.py
jno/utils/solver/sharding.py
jno/utils/solver/slip_runtime.py
jno/utils/solver/small_linalg.py
jno/utils/solver/solver_api.py
jno/utils/solver/solver_helper.py
jno/utils/solver/sparse_batching.py
jno/utils/solver/term_kind.py
jno/utils/solver/time_route.py
jno/utils/solver/timeschemes.py
jno/utils/solver/weak_form.py
jno/utils/solver/weak_form_helpers.py
tests/test_ad_mode.py
tests/test_adaptive.py
tests/test_adaptive_package.py
tests/test_bayesian.py
tests/test_checkpointing.py
tests/test_complex_vector_view.py
tests/test_compress_plan_collapse.py
tests/test_concat_temporal.py
tests/test_config.py
tests/test_core.py
tests/test_core_data_parallel.py
tests/test_core_device.py
tests/test_core_module_mutation.py
tests/test_core_parameter_only_domain.py
tests/test_cse.py
tests/test_curved_geometry.py
tests/test_derivatives.py
tests/test_doc_snippets.py
tests/test_domain_cell_aspect.py
tests/test_domain_cell_geometry.py
tests/test_domain_decomposition.py
tests/test_domain_export.py
tests/test_domain_geometries.py
tests/test_domain_load_mesh.py
tests/test_domain_load_msh.py
tests/test_domain_load_npz.py
tests/test_domain_meshfree.py
tests/test_domain_normals.py
tests/test_domain_tag.py
tests/test_dtypes.py
tests/test_encrypted_save_load.py
tests/test_engd_optimizer.py
tests/test_evaluator_derivative_reuse.py
tests/test_explainability.py
tests/test_factor_cache_block_count.py
tests/test_fdm.py
tests/test_fdm_spectral.py
tests/test_fdm_structured.py
tests/test_fem_1d.py
tests/test_fem_3d.py
tests/test_fem_adapt.py
tests/test_fem_adapt_1d.py
tests/test_fem_adapt_complex.py
tests/test_fem_adapt_constraint.py
tests/test_fem_adapt_criterion.py
tests/test_fem_adapt_enrich.py
tests/test_fem_adapt_multifield.py
tests/test_fem_adapt_quad.py
tests/test_fem_adapt_refine.py
tests/test_fem_adapt_space_guard.py
tests/test_fem_adapt_transient.py
tests/test_fem_adapt_transient_criterion.py
tests/test_fem_adaptive_load_path.py
tests/test_fem_adaptive_timestep.py
tests/test_fem_additive_terms.py
tests/test_fem_ale_mesh_velocity.py
tests/test_fem_ams_gradient.py
tests/test_fem_anisotropic_moving_mesh.py
tests/test_fem_arclength.py
tests/test_fem_argyris.py
tests/test_fem_assembly_on_host.py
tests/test_fem_axisymmetric.py
tests/test_fem_axisymmetric_vector.py
tests/test_fem_bdf2_state_dependent_mass.py
tests/test_fem_block_step.py
tests/test_fem_bounds.py
tests/test_fem_boussinesq.py
tests/test_fem_by_tag.py
tests/test_fem_callable_fields.py
tests/test_fem_cell_metric.py
tests/test_fem_cell_size.py
tests/test_fem_cellwise.py
tests/test_fem_chns_droplet.py
tests/test_fem_complex.py
tests/test_fem_complex_field.py
tests/test_fem_complex_fusion.py
tests/test_fem_complex_parametric.py
tests/test_fem_complex_single_assembly.py
tests/test_fem_contact_gap.py
tests/test_fem_contact_gear_ratio.py
tests/test_fem_contact_projection.py
tests/test_fem_contact_relax.py
tests/test_fem_contact_search.py
tests/test_fem_contact_slide.py
tests/test_fem_continuation.py
tests/test_fem_convergence.py
tests/test_fem_coupled.py
tests/test_fem_coupled_field_parameter.py
tests/test_fem_coupled_robustness.py
tests/test_fem_cover.py
tests/test_fem_cylinder_benchmark.py
tests/test_fem_dead_dofs.py
tests/test_fem_derived_coupling.py
tests/test_fem_dirichlet_parameters.py
tests/test_fem_dirichlet_row_scaling.py
tests/test_fem_dirichlet_symmetry.py
tests/test_fem_driver.py
tests/test_fem_drop_oscillation.py
tests/test_fem_droplets_merge.py
tests/test_fem_dynamic_topology.py
tests/test_fem_dynamic_topology_equivalence.py
tests/test_fem_eigs.py
tests/test_fem_eigs_nonsymmetric.py
tests/test_fem_element_operator.py
tests/test_fem_elements.py
tests/test_fem_enclosure_radiation.py
tests/test_fem_eval_readout.py
tests/test_fem_export.py
tests/test_fem_facet_connectivity.py
tests/test_fem_fieldsplit.py
tests/test_fem_finite_strain.py
tests/test_fem_follower_normals.py
tests/test_fem_free_surface_tension.py
tests/test_fem_frozen_parameters.py
tests/test_fem_frozen_readout.py
tests/test_fem_functional.py
tests/test_fem_galerkin_basis.py
tests/test_fem_geometry_law_reads_its_field.py
tests/test_fem_geometry_terms.py
tests/test_fem_geometry_terms_refuse_non_coordinates.py
tests/test_fem_geometry_terms_share_coordinates.py
tests/test_fem_graded_mesh.py
tests/test_fem_hanging_nodes.py
tests/test_fem_hermite.py
tests/test_fem_hessian.py
tests/test_fem_hex_tie.py
tests/test_fem_higher_order.py
tests/test_fem_history_march.py
tests/test_fem_interface_dirichlet.py
tests/test_fem_inverse.py
tests/test_fem_known_coefficient_field.py
tests/test_fem_les.py
tests/test_fem_linear_solvers.py
tests/test_fem_load_path_field.py
tests/test_fem_lsc.py
tests/test_fem_march_checkpoint.py
tests/test_fem_march_stats.py
tests/test_fem_matfun.py
tests/test_fem_morley.py
tests/test_fem_mortar.py
tests/test_fem_mortar_curved_3d.py
tests/test_fem_move_mesh.py
tests/test_fem_moving_mesh_remesh.py
tests/test_fem_native_lagrange.py
tests/test_fem_navier_stokes_3d.py
tests/test_fem_nedelec_3d.py
tests/test_fem_nedelec_anisotropic.py
tests/test_fem_nedelec_complex.py
tests/test_fem_nedelec_impedance.py
tests/test_fem_nedelec_incident.py
tests/test_fem_nedelec_periodic.py
tests/test_fem_neural_coefficients.py
tests/test_fem_newton_krylov.py
tests/test_fem_nonlinear_mass.py
tests/test_fem_nonnodal.py
tests/test_fem_nonnodal_assembly_memory.py
tests/test_fem_nonnodal_dsl.py
tests/test_fem_nonnodal_frozen_coefficient.py
tests/test_fem_nonnodal_mixed_lagrange.py
tests/test_fem_nonnodal_sparse_assembly.py
tests/test_fem_nonnodal_trainable_coords.py
tests/test_fem_operator_compression.py
tests/test_fem_ops.py
tests/test_fem_per_region.py
tests/test_fem_periodic_bloch.py
tests/test_fem_periodic_complex.py
tests/test_fem_periodic_complex_transient.py
tests/test_fem_periodic_coupling.py
tests/test_fem_periodic_multifield.py
tests/test_fem_periodic_parametric.py
tests/test_fem_periodic_transient.py
tests/test_fem_periodic_unstructured.py
tests/test_fem_phase_field_sent.py
tests/test_fem_pin.py
tests/test_fem_plate_bc.py
tests/test_fem_precond_algebra.py
tests/test_fem_precond_ams.py
tests/test_fem_promotion_entity_keys.py
tests/test_fem_quadhex_boundary_region.py
tests/test_fem_quadhex_domain.py
tests/test_fem_quadhex_element.py
tests/test_fem_quadhex_mesh.py
tests/test_fem_quadhex_solve.py
tests/test_fem_rebuild_cache.py
tests/test_fem_refine_hexes.py
tests/test_fem_refine_quads.py
tests/test_fem_refine_regions.py
tests/test_fem_region_coefficient_from_mesh.py
tests/test_fem_relocate.py
tests/test_fem_relocate_escalation.py
tests/test_fem_relocate_moving_mesh.py
tests/test_fem_relocate_objective.py
tests/test_fem_relocate_remesh.py
tests/test_fem_remesh_multimaterial.py
tests/test_fem_remesh_transfer.py
tests/test_fem_rosenbrock.py
tests/test_fem_saddle_generality.py
tests/test_fem_saddle_precond.py
tests/test_fem_saddle_warning.py
tests/test_fem_sdirk.py
tests/test_fem_second_order_time.py
tests/test_fem_sharding.py
tests/test_fem_sharding_newton.py
tests/test_fem_sharding_transient.py
tests/test_fem_slip_bc.py
tests/test_fem_slip_runtime_coords.py
tests/test_fem_solve_values.py
tests/test_fem_solver_amg.py
tests/test_fem_solver_amg_jaxamg.py
tests/test_fem_solver_krylov.py
tests/test_fem_solver_picard.py
tests/test_fem_solver_precond.py
tests/test_fem_solver_precond_cached.py
tests/test_fem_solver_slots.py
tests/test_fem_solver_transient.py
tests/test_fem_space_api.py
tests/test_fem_sparse_assembly.py
tests/test_fem_stabilised_flow.py
tests/test_fem_staggered.py
tests/test_fem_staggered_groups.py
tests/test_fem_state_region.py
tests/test_fem_stokes_3d.py
tests/test_fem_tensor_ops.py
tests/test_fem_term_kind.py
tests/test_fem_tie_dirichlet_conflict.py
tests/test_fem_time_schemes.py
tests/test_fem_topology.py
tests/test_fem_trainable_coordinates.py
tests/test_fem_trainable_normals.py
tests/test_fem_trainable_placeholder.py
tests/test_fem_trainables_gaps.py
tests/test_fem_transfer_solution.py
tests/test_fem_transient_bc.py
tests/test_fem_transient_convergence_guard.py
tests/test_fem_vector.py
tests/test_fem_vector_components.py
tests/test_fem_vector_tie.py
tests/test_fem_volume_region_guard.py
tests/test_fem_workflow.py
tests/test_field_view.py
tests/test_fn_adaptive.py
tests/test_fn_pdes.py
tests/test_foundax_eqx.py
tests/test_geometry_shape.py
tests/test_gmsh_quiet.py
tests/test_grf.py
tests/test_history_index.py
tests/test_hybrid_nn_fem.py
tests/test_info.py
tests/test_info_breadth.py
tests/test_info_dd.py
tests/test_integration.py
tests/test_integration_operators.py
tests/test_iree.py
tests/test_jno_optimizers.py
tests/test_laplacian_fusion.py
tests/test_lattice_mg.py
tests/test_lattice_probe.py
tests/test_lazy_sources.py
tests/test_logger.py
tests/test_lora.py
tests/test_march_hoist.py
tests/test_march_no_op.py
tests/test_matvec_format.py
tests/test_md_decouple_optimizer.py
tests/test_mesh_cell_region_membership.py
tests/test_mesh_nodal_measures.py
tests/test_model_controls_attachment.py
tests/test_models.py
tests/test_n1e_reconstruction_3d.py
tests/test_named_axes.py
tests/test_newton_assembled_default.py
tests/test_newton_convergence_guard.py
tests/test_newton_reuse.py
tests/test_node_eval.py
tests/test_noise.py
tests/test_nondim.py
tests/test_numpy.py
tests/test_one_language.py
tests/test_operator_composition.py
tests/test_operator_placement.py
tests/test_operators_convergence.py
tests/test_operators_mms.py
tests/test_operators_mms_3d.py
tests/test_operators_pipeline.py
tests/test_optics_beam_depth.py
tests/test_optimizer_state_persists.py
tests/test_optimizers.py
tests/test_optimizers_mma.py
tests/test_orphan_nodes.py
tests/test_packaging.py
tests/test_pdeformer2_bridge.py
tests/test_periodic_parametric_integration.py
tests/test_plasticity.py
tests/test_polygon_domain_build_mesh.py
tests/test_polygon_domain_csg.py
tests/test_polygon_domain_regions.py
tests/test_polygon_domain_time.py
tests/test_precond_amg_on_a_march.py
tests/test_precond_ams_mixed_build.py
tests/test_precond_float32.py
tests/test_precond_form_inner.py
tests/test_precond_fsai.py
tests/test_precond_hypre.py
tests/test_precond_ilu.py
tests/test_precond_inner_precond.py
tests/test_precond_real_equivalent.py
tests/test_precond_schwarz.py
tests/test_profile.py
tests/test_rcwa.py
tests/test_rcwa_aerial.py
tests/test_rcwa_anisotropic.py
tests/test_rcwa_general_anisotropic.py
tests/test_rcwa_incidence.py
tests/test_rcwa_pml.py
tests/test_rcwa_profile.py
tests/test_rcwa_smoothing.py
tests/test_rcwa_source.py
tests/test_rcwa_vector.py
tests/test_reconnect_alpha.py
tests/test_reconnect_graded.py
tests/test_regularize.py
tests/test_resampling.py
tests/test_resampling_jit.py
tests/test_scale_dynamic_lr.py
tests/test_scheme_families.py
tests/test_shape_attach.py
tests/test_shape_boundary_traced.py
tests/test_shape_contains.py
tests/test_shape_nonconforming.py
tests/test_shape_nonconforming_multiface.py
tests/test_shape_regions.py
tests/test_shape_sample_traced.py
tests/test_shape_sampling.py
tests/test_shape_sdf.py
tests/test_shape_structured.py
tests/test_shape_tessellation.py
tests/test_simp_continuation.py
tests/test_solver_lu_backends.py
tests/test_sparse_batching.py
tests/test_sparse_matvec.py
tests/test_spectral_scheme.py
tests/test_stencils.py
tests/test_symbol_positional_time.py
tests/test_tag_geometry_name_collision.py
tests/test_tag_region_surfaces.py
tests/test_tensor_layout.py
tests/test_topology_optimisation.py
tests/test_trace.py
tests/test_trace_compiler.py
tests/test_trace_evaluator.py
tests/test_trace_unary_plus.py
tests/test_transient_march_cache.py
tests/test_transient_step_slot_operator.py
tests/test_tune_sequence.py
tests/test_tuner.py
tests/test_tutorial_examples.py
tests/test_units.py
tests/test_view_factor_closed.py
tests/test_views.py
tests/test_vpinn_integration.py
tests/test_wandb_integration.py
tests/test_weak_form_scalar_distribution.py
tests/test_workflow_correctness.py
tests/test_x64_isolation.py