LICENSE
MANIFEST.in
README.md
pyproject.toml
mojogp/__init__.py
mojogp/_multi_output_backend.py
mojogp/_provider_lifecycle.py
mojogp/_provider_runtime.py
mojogp/_routes.py
mojogp/_version.py
mojogp/compiler.py
mojogp/feature_support.py
mojogp/gp.py
mojogp/kernel.py
mojogp/loader.py
mojogp/multi_output_gp.py
mojogp/pathwise_prior.py
mojogp/progress.py
mojogp/settings.py
mojogp/utils.py
mojogp.egg-info/PKG-INFO
mojogp.egg-info/SOURCES.txt
mojogp.egg-info/dependency_links.txt
mojogp.egg-info/requires.txt
mojogp.egg-info/top_level.txt
mojogp/benchmark/__init__.py
mojogp/benchmark/config.py
mojogp/codegen_engine/__init__.py
mojogp/codegen_engine/compiler.py
mojogp/codegen_engine/differentiation.py
mojogp/codegen_engine/expressions.py
mojogp/codegen_engine/ir.py
mojogp/codegen_engine/overrides.py
mojogp/codegen_engine/registry.py
mojogp/codegen_engine/schedule.py
mojogp/codegen_engine/emit/__init__.py
mojogp/codegen_engine/emit/builder.py
mojogp/codegen_engine/emit/fn_ptr_module.py
mojogp/codegen_engine/emit/gpu_kernel.py
mojogp/codegen_engine/emit/module.py
mojogp/codegen_engine/emit/mojo_printer.py
mojogp/codegen_engine/passes/__init__.py
mojogp/codegen_engine/passes/cse.py
mojogp/codegen_engine/passes/dead_code.py
mojogp/codegen_engine/passes/forward_hoist.py
mojogp/codegen_engine/passes/gradient_sharing.py
mojogp/codegen_engine/passes/inv_ls.py
mojogp/codegen_engine/passes/numerical_stability.py
mojogp/codegen_engine/passes/simplify.py
mojogp/codegen_engine/passes/strength.py
mojogp/kernels/__init__.mojo
mojogp/kernels/bbmm_gpu_kernels.mojo
mojogp/kernels/bbmm_types.mojo
mojogp/kernels/categorical_composable.mojo
mojogp/kernels/categorical_kernel.mojo
mojogp/kernels/categorical_state.mojo
mojogp/kernels/cg_solver.mojo
mojogp/kernels/cg_with_provider.mojo
mojogp/kernels/combined_inv_quad_logdet.mojo
mojogp/kernels/composable_kernel.mojo
mojogp/kernels/composite_matvec.mojo
mojogp/kernels/composite_prediction.mojo
mojogp/kernels/composite_provider.mojo
mojogp/kernels/constants.mojo
mojogp/kernels/cross_covariance_provider.mojo
mojogp/kernels/gemm_matvec.mojo
mojogp/kernels/gemm_tiled.mojo
mojogp/kernels/generic_matvec.mojo
mojogp/kernels/gpu_utils.mojo
mojogp/kernels/gradient_provider.mojo
mojogp/kernels/kernel_functions.mojo
mojogp/kernels/kernel_materialization.mojo
mojogp/kernels/kernel_params.mojo
mojogp/kernels/kronecker_direct_provider.mojo
mojogp/kernels/kronecker_gpu_kernels.mojo
mojogp/kernels/kronecker_preconditioner.mojo
mojogp/kernels/kronecker_provider.mojo
mojogp/kernels/lanczos.mojo
mojogp/kernels/lanczos_with_provider.mojo
mojogp/kernels/lmc_bbmm_step.mojo
mojogp/kernels/lmc_bbmm_step_mf.mojo
mojogp/kernels/lmc_composite_provider.mojo
mojogp/kernels/lmc_preconditioner.mojo
mojogp/kernels/lmc_provider.mojo
mojogp/kernels/love_provider.mojo
mojogp/kernels/matvec_provider.mojo
mojogp/kernels/mixed_composite_matvec.mojo
mojogp/kernels/mixed_composite_provider.mojo
mojogp/kernels/mixed_composite_training.mojo
mojogp/kernels/multi_output_prediction.mojo
mojogp/kernels/native_numerics.mojo
mojogp/kernels/pivoted_cholesky.mojo
mojogp/kernels/preconditioner_trait.mojo
mojogp/kernels/profiling_config.mojo
mojogp/kernels/py_conversion.mojo
mojogp/kernels/task_covariance.mojo
mojogp/kernels/training.mojo
mojogp/kernels/training_types.mojo
mojogp/kernels/training_utils.mojo
mojogp/kernels/utils.mojo
mojogp/kernels/jit/__init__.mojo
mojogp/kernels/jit/erased_provider.mojo
mojogp/kernels/jit/fused_kronecker_provider.mojo
mojogp/kernels/jit/jit_categorical_params.mojo
mojogp/kernels/jit/jit_engine_binding_helpers.mojo
mojogp/kernels/jit/jit_engine_bindings.mojo
mojogp/kernels/jit/jit_engine_lmc_bindings.mojo
mojogp/kernels/jit/jit_engine_multi_output_bindings.mojo
mojogp/kernels/jit/jit_lmc.mojo
mojogp/kernels/jit/jit_lmc_mixed.mojo
mojogp/kernels/jit/jit_mixed.mojo
mojogp/kernels/jit/jit_multi_output.mojo
mojogp/kernels/jit/jit_multi_output_mixed.mojo
mojogp/kernels/jit/jit_prediction.mojo
mojogp/kernels/jit/jit_progress.mojo
mojogp/kernels/jit/jit_training.mojo
mojogp/specialization/__init__.py
mojogp/specialization/benchmark.py
mojogp/specialization/builtins.py
mojogp/specialization/decision.py
mojogp/specialization/descriptor.py
mojogp/specialization/profile.py
mojogp/specialization/registry.py
mojogp/specialization/rules.py
mojogp/specialization/translate.py