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/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/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/engd.py
jno/optimizers/md_decouple.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/profiling.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/backend_blocks.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_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_route.py
jno/utils/solver/fem_topology.py
jno/utils/solver/fem_utils.py
jno/utils/solver/geometric_mg.py
jno/utils/solver/history_march.py
jno/utils/solver/krylov.py
jno/utils/solver/linear.py
jno/utils/solver/mass.py
jno/utils/solver/matfun.py
jno/utils/solver/newton_krylov.py
jno/utils/solver/parametric_helpers.py
jno/utils/solver/sharding.py
jno/utils/solver/solver_api.py
jno/utils/solver/solver_helper.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_concat_temporal.py
tests/test_config.py
tests/test_core.py
tests/test_core_device.py
tests/test_curved_geometry.py
tests/test_derivatives.py
tests/test_doc_snippets.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_npz.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_fdm.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_transient.py
tests/test_fem_adaptive_timestep.py
tests/test_fem_ams_gradient.py
tests/test_fem_argyris.py
tests/test_fem_axisymmetric.py
tests/test_fem_block_step.py
tests/test_fem_boussinesq.py
tests/test_fem_callable_fields.py
tests/test_fem_cell_size.py
tests/test_fem_complex.py
tests/test_fem_complex_field.py
tests/test_fem_complex_parametric.py
tests/test_fem_contact_gap.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_driver.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_facet_connectivity.py
tests/test_fem_finite_strain.py
tests/test_fem_frozen_parameters.py
tests/test_fem_frozen_readout.py
tests/test_fem_galerkin_basis.py
tests/test_fem_geometry_terms.py
tests/test_fem_hermite.py
tests/test_fem_hessian.py
tests/test_fem_higher_order.py
tests/test_fem_history_march.py
tests/test_fem_inverse.py
tests/test_fem_linear_solvers.py
tests/test_fem_load_path_field.py
tests/test_fem_matfun.py
tests/test_fem_morley.py
tests/test_fem_mortar.py
tests/test_fem_move_mesh.py
tests/test_fem_native_lagrange.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_dsl.py
tests/test_fem_nonnodal_sparse_assembly.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_pin.py
tests/test_fem_plate_bc.py
tests/test_fem_precond_ams.py
tests/test_fem_relocate.py
tests/test_fem_second_order_time.py
tests/test_fem_sharding.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_term_kind.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_transfer_solution.py
tests/test_fem_transient_bc.py
tests/test_fem_vector.py
tests/test_fem_vector_components.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_geometric_mg.py
tests/test_geometry_shape.py
tests/test_history_index.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_logger.py
tests/test_lora.py
tests/test_md_decouple_optimizer.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_newton_convergence_guard.py
tests/test_node_eval.py
tests/test_noise.py
tests/test_nondim.py
tests/test_numpy.py
tests/test_operators_convergence.py
tests/test_operators_mms.py
tests/test_operators_mms_3d.py
tests/test_operators_pipeline.py
tests/test_optimizers.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_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_regularize.py
tests/test_resampling.py
tests/test_scale_dynamic_lr.py
tests/test_shape_contains.py
tests/test_shape_nonconforming.py
tests/test_shape_regions.py
tests/test_solver_lu_backends.py
tests/test_symbol_positional_time.py
tests/test_trace.py
tests/test_trace_compiler.py
tests/test_trace_evaluator.py
tests/test_trace_unary_plus.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_workflow_correctness.py