LICENSE
README.md
pyproject.toml
jax_neural_operators.egg-info/PKG-INFO
jax_neural_operators.egg-info/SOURCES.txt
jax_neural_operators.egg-info/dependency_links.txt
jax_neural_operators.egg-info/requires.txt
jax_neural_operators.egg-info/top_level.txt
jno/__init__.py
jno/core.py
jno/differential_operators.py
jno/fn.py
jno/integration_operators.py
jno/jnp_ops.py
jno/lora.py
jno/noise.py
jno/py.typed
jno/trace.py
jno/trace_compiler.py
jno/trace_evaluator.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/geometries.py
jno/domain/mesh_utils.py
jno/domain/meshio_mixin.py
jno/domain/polygon_domain.py
jno/utils/__init__.py
jno/utils/config.py
jno/utils/explainability.py
jno/utils/iree.py
jno/utils/load_save.py
jno/utils/logger.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/backend_blocks.py
jno/utils/solver/feax_utils.py
jno/utils/solver/fem_route.py
jno/utils/solver/solver_helper.py
jno/utils/solver/time_adapters.py
jno/utils/solver/time_route.py
jno/utils/solver/weak_form.py
jno/utils/solver/weak_form_helpers.py
tests/test_adaptive.py
tests/test_adaptive_package.py
tests/test_checkpointing.py
tests/test_config.py
tests/test_core.py
tests/test_core_device.py
tests/test_derivatives.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_encrypted_save_load.py
tests/test_explainability.py
tests/test_fem_integration.py
tests/test_fn_adaptive.py
tests/test_fn_pdes.py
tests/test_foundax_eqx.py
tests/test_integration.py
tests/test_integration_operators.py
tests/test_iree.py
tests/test_logger.py
tests/test_lora.py
tests/test_model_controls_attachment.py
tests/test_models.py
tests/test_noise.py
tests/test_numpy.py
tests/test_packaging.py
tests/test_pdeformer2_bridge.py
tests/test_polygon_domain_csg.py
tests/test_resampling.py
tests/test_time_route_integration.py
tests/test_trace.py
tests/test_trace_compiler.py
tests/test_trace_evaluator.py
tests/test_tuner.py
tests/test_tutorial_examples_smoke.py
tests/test_vpinn_integration.py
tests/test_wandb_integration.py
tests/test_workflow_correctness.py