CHANGELOG.md
LICENSE
MANIFEST.in
README.md
pyproject.toml
neuralops/__init__.py
neuralops/__main__.py
neuralops/api.py
neuralops/main.py
neuralops/py.typed
neuralops/recipes.py
neuralops/transformers.py
neuralops.egg-info/PKG-INFO
neuralops.egg-info/SOURCES.txt
neuralops.egg-info/dependency_links.txt
neuralops.egg-info/entry_points.txt
neuralops.egg-info/requires.txt
neuralops.egg-info/top_level.txt
neuralops/analysis/__init__.py
neuralops/analysis/generate_all_modules_analysis.py
neuralops/analysis/generate_sweep_publication_artifacts.py
neuralops/analysis/inspector.py
neuralops/analysis/layerwise_analyzer.py
neuralops/analysis/metrics.py
neuralops/analysis/model_comparison.py
neuralops/analysis/paper_plots.py
neuralops/analysis/plotting.py
neuralops/analysis/report_generator.py
neuralops/analysis/utils.py
neuralops/analysis/comparison/__init__.py
neuralops/analysis/comparison/statistical_tests.py
neuralops/analysis/module_analysis/__init__.py
neuralops/analysis/module_analysis/aggregators.py
neuralops/analysis/module_analysis/constants.py
neuralops/analysis/module_analysis/main.py
neuralops/analysis/module_analysis/parsers.py
neuralops/analysis/module_analysis/reporters.py
neuralops/attention/__init__.py
neuralops/attentions/__init__.py
neuralops/attentions/adaptive/__init__.py
neuralops/attentions/adaptive/adaptive_span_attention.py
neuralops/attentions/adaptive/block_transformer_hiba.py
neuralops/attentions/adaptive/eat.py
neuralops/attentions/adaptive/infini_attention.py
neuralops/attentions/adaptive/lola_attention.py
neuralops/attentions/adaptive/progressive_token_pruning.py
neuralops/attentions/adaptive/sink_attention.py
neuralops/attentions/adaptive/span_expanded_attention.py
neuralops/attentions/adaptive/squeezing_heads_distill.py
neuralops/attentions/adaptive/token_compressed_attention.py
neuralops/attentions/dense/__init__.py
neuralops/attentions/dense/full_attention.py
neuralops/attentions/dense/zeros_attention.py
neuralops/attentions/experimental/__init__.py
neuralops/attentions/experimental/template_attention.py
neuralops/attentions/fast_kernels/__init__.py
neuralops/attentions/fast_kernels/adasplash_attention.py
neuralops/attentions/fast_kernels/flash_attention3.py
neuralops/attentions/fast_kernels/flash_attention3_fp8.py
neuralops/attentions/fast_kernels/soa_attn.py
neuralops/attentions/fast_kernels/xattention.py
neuralops/attentions/gated/__init__.py
neuralops/attentions/gated/decay_linear_attention.py
neuralops/attentions/gated/flagship_fla.py
neuralops/attentions/gated/gated_delta_net.py
neuralops/attentions/gated/gated_deltanet_block.py
neuralops/attentions/gated/hgrn2_gated_deltanet.py
neuralops/attentions/gated/hgrn2_gated_layer.py
neuralops/attentions/gated/hgrn2_gated_outer.py
neuralops/attentions/gated/hla_attention.py
neuralops/attentions/gated/kda_attention.py
neuralops/attentions/hybrid/__init__.py
neuralops/attentions/hybrid/fusion_attn.py
neuralops/attentions/hybrid/gated_hybrid_kernel_attention.py
neuralops/attentions/hybrid/granite_hybrid.py
neuralops/attentions/hybrid/halo_regla.py
neuralops/attentions/hybrid/hybrid_attention.py
neuralops/attentions/hybrid/hybrid_local_kernel_attention.py
neuralops/attentions/hybrid/landmark_local_attention.py
neuralops/attentions/hybrid/mixattention_moska.py
neuralops/attentions/hybrid/optimized_hybrid_attention.py
neuralops/attentions/hybrid/positive_kernel_linear_attention.py
neuralops/attentions/hybrid/refined_landmark_attention.py
neuralops/attentions/hybrid/regla_attention.py
neuralops/attentions/hybrid/spectron_fp8_attention.py
neuralops/attentions/hybrid/split_head_hybrid_attention.py
neuralops/attentions/hybrid/tri_branch_hybrid_attention.py
neuralops/attentions/linear/__init__.py
neuralops/attentions/linear/chela_slay.py
neuralops/attentions/linear/cosformer.py
neuralops/attentions/linear/galerkin_attention.py
neuralops/attentions/linear/gla_attention.py
neuralops/attentions/linear/linformer_attention.py
neuralops/attentions/linear/nasc_attention.py
neuralops/attentions/linear/nystrom_attention.py
neuralops/attentions/linear/performer_attention.py
neuralops/attentions/linear/primal_attention.py
neuralops/attentions/linear/rala_attention.py
neuralops/attentions/linear/rpc_attention.py
neuralops/attentions/linear/spectra_v2.py
neuralops/attentions/linear/spectraformer_attention.py
neuralops/attentions/linear/spectron.py
neuralops/attentions/linear/tlinformer_attention.py
neuralops/attentions/linear/tpa_attention.py
neuralops/attentions/linear/triton_cosformer.py
neuralops/attentions/modern/__init__.py
neuralops/attentions/modern/alibi.py
neuralops/attentions/modern/cross_attention.py
neuralops/attentions/modern/dynamic_conv_attention.py
neuralops/attentions/modern/gqa_attention.py
neuralops/attentions/modern/mla_attention.py
neuralops/attentions/modern/mqa.py
neuralops/attentions/modern/nha_attention.py
neuralops/attentions/modern/qcqa_attention.py
neuralops/attentions/modern/unified_rope.py
neuralops/attentions/modern/vig_attention.py
neuralops/attentions/modern/wgqa_attention.py
neuralops/attentions/sparse/__init__.py
neuralops/attentions/sparse/adaptive_sparse_attention.py
neuralops/attentions/sparse/bam_umoe.py
neuralops/attentions/sparse/cluster_attention.py
neuralops/attentions/sparse/expert_choice_moe.py
neuralops/attentions/sparse/hamming_attention.py
neuralops/attentions/sparse/lsh_attention.py
neuralops/attentions/sparse/moba_attention.py
neuralops/attentions/sparse/moe_deltakv_paged.py
neuralops/attentions/sparse/nsa_attention.py
neuralops/attentions/sparse/padre_attention.py
neuralops/attentions/sparse/reap_moe.py
neuralops/attentions/sparse/routing_attention.py
neuralops/attentions/ssm/__init__.py
neuralops/attentions/ssm/goldfinch_ahn.py
neuralops/attentions/ssm/griffin_ssd.py
neuralops/attentions/ssm/hymba_transx.py
neuralops/attentions/ssm/jamba.py
neuralops/attentions/ssm/jamba_info.py
neuralops/attentions/ssm/mamba_block.py
neuralops/attentions/ssm/mamba_svd_flash.py
neuralops/attentions/ssm/nexus.py
neuralops/attentions/ssm/orion.py
neuralops/attentions/ssm/phantom.py
neuralops/attentions/ssm/samba_m3.py
neuralops/attentions/ssm/samba_moe.py
neuralops/attentions/ssm/selective_ssm.py
neuralops/attentions/ssm/simple_ssm.py
neuralops/attentions/ssm/spectral_mamba.py
neuralops/attentions/ssm/upi_stuffed_mamba.py
neuralops/attentions/ssm/vspec.py
neuralops/attentions/svd/__init__.py
neuralops/attentions/svd/adaptive_svd_attention.py
neuralops/attentions/svd/asvd_attention.py
neuralops/attentions/svd/compressed_attention.py
neuralops/attentions/svd/compressed_mha_attention.py
neuralops/attentions/svd/factorised_matrix_state.py
neuralops/attentions/svd/fused_svd_kernel.py
neuralops/attentions/svd/lora_attention.py
neuralops/attentions/svd/minimal_nova.py
neuralops/attentions/svd/minimal_svd.py
neuralops/attentions/svd/naive_svd.py
neuralops/attentions/svd/projection_factorized_attn.py
neuralops/attentions/svd/residual_svd_attention.py
neuralops/attentions/svd/salt_attention.py
neuralops/attentions/svd/svd_attention.py
neuralops/attentions/svd/svd_compressed.py
neuralops/attentions/svd/svd_linear_attention.py
neuralops/attentions/svd/svd_linear_decay.py
neuralops/attentions/svd/svd_llm_v2_attention.py
neuralops/attentions/svd/svd_mamba.py
neuralops/attentions/svd/svd_projection_attention.py
neuralops/attentions/svd/triton_svd_linear_attention.py
neuralops/benchmarks/__init__.py
neuralops/benchmarks/__main__.py
neuralops/benchmarks/api.py
neuralops/benchmarks/cli.py
neuralops/benchmarks/config.py
neuralops/benchmarks/discovery.py
neuralops/benchmarks/discovery_overrides.json
neuralops/benchmarks/eval_suite.py
neuralops/benchmarks/executor.py
neuralops/benchmarks/gpu_live_probe.py
neuralops/benchmarks/harvest_gpu_benchmark_data.py
neuralops/benchmarks/kernel_benchmark.py
neuralops/benchmarks/loggers.py
neuralops/benchmarks/logging.py
neuralops/benchmarks/orchestrator.py
neuralops/benchmarks/pareto_paper_suite.py
neuralops/benchmarks/registry.py
neuralops/benchmarks/result.py
neuralops/benchmarks/run_gpt2_sweep.py
neuralops/benchmarks/run_quality_sweep.py
neuralops/benchmarks/runner.py
neuralops/benchmarks/static_catalog.json
neuralops/benchmarks/suite_runner.py
neuralops/benchmarks/writers.py
neuralops/benchmarks/adapters/__init__.py
neuralops/benchmarks/adapters/classification.py
neuralops/benchmarks/adapters/perplexity.py
neuralops/benchmarks/adapters/ruler.py
neuralops/benchmarks/adapters/zeroshot.py
neuralops/benchmarks/core/__init__.py
neuralops/benchmarks/core/arguments.py
neuralops/benchmarks/core/data_loader.py
neuralops/benchmarks/core/evaluation.py
neuralops/benchmarks/core/kd.py
neuralops/benchmarks/core/model_loader.py
neuralops/benchmarks/core/stats.py
neuralops/benchmarks/core/training.py
neuralops/benchmarks/data/__init__.py
neuralops/benchmarks/data/loader.py
neuralops/benchmarks/kd/__init__.py
neuralops/benchmarks/kd/__main__.py
neuralops/benchmarks/kd/config.py
neuralops/benchmarks/kd/reporter.py
neuralops/benchmarks/kd/runner.py
neuralops/benchmarks/metrics/__init__.py
neuralops/benchmarks/metrics/evaluator.py
neuralops/benchmarks/models/__init__.py
neuralops/benchmarks/models/loader.py
neuralops/benchmarks/results/__init__.py
neuralops/benchmarks/results/dedup.py
neuralops/benchmarks/results/git_sync.py
neuralops/benchmarks/results/hf_sync.py
neuralops/benchmarks/results/index.py
neuralops/benchmarks/results/reproduce.py
neuralops/benchmarks/results/saver.py
neuralops/benchmarks/results/schema.py
neuralops/benchmarks/results/summarizer.py
neuralops/benchmarks/tasks/adapter.py
neuralops/benchmarks/utils/__init__.py
neuralops/benchmarks/utils/gpu_probe.py
neuralops/benchmarks/utils/harvest_data.py
neuralops/cli/__init__.py
neuralops/cli/bench_cmd.py
neuralops/cli/compress_cmd.py
neuralops/cli/doctor.py
neuralops/cli/eval_cmd.py
neuralops/cli/info.py
neuralops/cli/list_cmd.py
neuralops/cli/serve_cmd.py
neuralops/cli/terminal_ui.py
neuralops/cli/test_cmd.py
neuralops/cli/train_cmd.py
neuralops/compression/__init__.py
neuralops/compression_engine/__init__.py
neuralops/compression_engine/experimental/__init__.py
neuralops/compression_engine/experimental/flat_llm.py
neuralops/compression_engine/experimental/hashed_net_compressor.py
neuralops/compression_engine/experimental/liftr.py
neuralops/compression_engine/experimental/lillama_compressor.py
neuralops/compression_engine/experimental/roast_hashed_linear.py
neuralops/compression_engine/kv_cache/__init__.py
neuralops/compression_engine/kv_cache/aatc_xkv_compressor.py
neuralops/compression_engine/kv_cache/cachenotes_compressor.py
neuralops/compression_engine/kv_cache/delta_kv.py
neuralops/compression_engine/kv_cache/hybridkv_compressor.py
neuralops/compression_engine/kv_cache/kvtc_compressor.py
neuralops/compression_engine/kv_cache/zerokv_neo_compressor.py
neuralops/compression_engine/low_rank/__init__.py
neuralops/compression_engine/low_rank/a3_compressor.py
neuralops/compression_engine/low_rank/come_compressor.py
neuralops/compression_engine/low_rank/compot_compressor.py
neuralops/compression_engine/low_rank/cr_compressor.py
neuralops/compression_engine/low_rank/cur_compressor.py
neuralops/compression_engine/low_rank/cur_moe_compressor.py
neuralops/compression_engine/low_rank/dora_qlora.py
neuralops/compression_engine/low_rank/drone_compressor.py
neuralops/compression_engine/low_rank/enlighten_compressor.py
neuralops/compression_engine/low_rank/frandi_compressor.py
neuralops/compression_engine/low_rank/grecodec_compressor.py
neuralops/compression_engine/low_rank/lsc_compressor.py
neuralops/compression_engine/low_rank/mgie_svd_compressor.py
neuralops/compression_engine/low_rank/mlorq_compressor.py
neuralops/compression_engine/low_rank/modegpt_compressor.py
neuralops/compression_engine/low_rank/nmf_compressor.py
neuralops/compression_engine/low_rank/oats_compressor.py
neuralops/compression_engine/low_rank/rpca_compressor.py
neuralops/compression_engine/low_rank/salr.py
neuralops/compression_engine/low_rank/sdrm_compressor.py
neuralops/compression_engine/low_rank/slim_compressor.py
neuralops/compression_engine/low_rank/soka_compressor.py
neuralops/compression_engine/low_rank/svd_llm_compressor.py
neuralops/compression_engine/low_rank/tomoe_compressor.py
neuralops/compression_engine/low_rank/uniql_compressor.py
neuralops/compression_engine/pruning/__init__.py
neuralops/compression_engine/pruning/admm_compressor.py
neuralops/compression_engine/pruning/admm_pruner.py
neuralops/compression_engine/pruning/armor_pruner.py
neuralops/compression_engine/pruning/ast_pruner.py
neuralops/compression_engine/pruning/bonsai_pruner.py
neuralops/compression_engine/pruning/flap.py
neuralops/compression_engine/pruning/mamba_shedder.py
neuralops/compression_engine/pruning/proxsparse.py
neuralops/compression_engine/pruning/sparsegpt.py
neuralops/compression_engine/pruning/wanda.py
neuralops/compression_engine/quantization/__init__.py
neuralops/compression_engine/quantization/astro_quantizer.py
neuralops/compression_engine/quantization/awq.py
neuralops/compression_engine/quantization/coopq_quantizer.py
neuralops/compression_engine/quantization/faq_quantizer.py
neuralops/compression_engine/quantization/gptq.py
neuralops/compression_engine/quantization/huffman_coder.py
neuralops/compression_engine/quantization/kmeans_quantize.py
neuralops/compression_engine/quantization/rotational_slicegpt_int4.py
neuralops/compression_engine/quantization/slicegpt.py
neuralops/compression_engine/quantization/slicegpt_pca.py
neuralops/compression_engine/quantization/smoothquant.py
neuralops/compression_engine/svd/__init__.py
neuralops/compression_engine/svd/aa_svd_compressor.py
neuralops/compression_engine/svd/adaptive_ranks.py
neuralops/compression_engine/svd/adasvd_compressor.py
neuralops/compression_engine/svd/auto_svd.py
neuralops/compression_engine/svd/direct_svd.py
neuralops/compression_engine/svd/dobi_svd.py
neuralops/compression_engine/svd/dynamic_rank.py
neuralops/compression_engine/svd/entropy_budget_allocation.py
neuralops/compression_engine/svd/erc_svd.py
neuralops/compression_engine/svd/fwsvd_compressor.py
neuralops/compression_engine/svd/gfws_d_compressor.py
neuralops/compression_engine/svd/hadamard_incoherent_svd.py
neuralops/compression_engine/svd/hybrid_svd_layer.py
neuralops/compression_engine/svd/optimal_subspace_projection.py
neuralops/compression_engine/svd/residual_svd_compressor.py
neuralops/compression_engine/svd/saes_svd_v2.py
neuralops/compression_engine/svd/subspace_projection.py
neuralops/compression_engine/svd/svd_linear.py
neuralops/compression_engine/svd/swift_svd.py
neuralops/compression_engine/svd/swift_svd_act.py
neuralops/compression_engine/tensor_decomp/__init__.py
neuralops/compression_engine/tensor_decomp/kronecker_compressor.py
neuralops/compression_engine/tensor_decomp/kronecker_tt_lora.py
neuralops/compression_engine/tensor_decomp/tt_compressor.py
neuralops/config/__init__.py
neuralops/config/compression_config.py
neuralops/config/config.py
neuralops/config/data_config.py
neuralops/config/experiment_config.py
neuralops/config/hardware_config.py
neuralops/config/loader.py
neuralops/config/method_config.py
neuralops/config/model_config.py
neuralops/config/sweep_config.py
neuralops/config/training_config.py
neuralops/core/__init__.py
neuralops/core/_attention_extractors.py
neuralops/core/_attention_helpers.py
neuralops/core/_attention_injectors.py
neuralops/core/_attention_surgery.py
neuralops/core/_attention_wrapper.py
neuralops/core/adapters.py
neuralops/core/amp_utils.py
neuralops/core/architecture.py
neuralops/core/attention_mask.py
neuralops/core/attention_utils.py
neuralops/core/base.py
neuralops/core/cache_utils.py
neuralops/core/common.py
neuralops/core/composable.py
neuralops/core/compression_engine.py
neuralops/core/compression_pipeline.py
neuralops/core/config.py
neuralops/core/cuda_graph.py
neuralops/core/deprecated_aliases.json
neuralops/core/exceptions.py
neuralops/core/fused_norm_attention.py
neuralops/core/gqa_utils.py
neuralops/core/inspector.py
neuralops/core/lazy_import.py
neuralops/core/legacy_aliases.py
neuralops/core/mask_utils.py
neuralops/core/memory.py
neuralops/core/metrics.py
neuralops/core/model_family.py
neuralops/core/model_utils.py
neuralops/core/model_wrapper.py
neuralops/core/paper_export.py
neuralops/core/pruning_utils.py
neuralops/core/quantized_attention.py
neuralops/core/registry.py
neuralops/core/replace_attention.py
neuralops/core/roofline.py
neuralops/core/rope.py
neuralops/core/run_long_context_sweep.py
neuralops/core/selective_scan.py
neuralops/core/snapshot_restore.py
neuralops/core/svd_utils.py
neuralops/core/t5_compat.py
neuralops/core/triton_linear_attn.py
neuralops/core/triton_sparse_attn.py
neuralops/core/utils.py
neuralops/core/autograd/__init__.py
neuralops/core/autograd/flash_attention.py
neuralops/core/autograd/fused_backward.py
neuralops/core/backend/__init__.py
neuralops/core/backend/dispatch.py
neuralops/core/fused/__init__.py
neuralops/core/fused/fused_ce.py
neuralops/core/fused/fused_kd.py
neuralops/core/fused/fused_norm_linear.py
neuralops/core/triton_kernels/__init__.py
neuralops/core/triton_kernels/_common.py
neuralops/core/triton_kernels/_types.py
neuralops/core/triton_kernels/autotuner.py
neuralops/core/triton_kernels/block_sparse.py
neuralops/core/triton_kernels/cosformer_triton_kernel.py
neuralops/core/triton_kernels/elementwise.py
neuralops/core/triton_kernels/flash_decoding.py
neuralops/core/triton_kernels/fms_scan.py
neuralops/core/triton_kernels/fused_ce.py
neuralops/core/triton_kernels/fused_kd.py
neuralops/core/triton_kernels/fused_norm_linear.py
neuralops/core/triton_kernels/fused_rope.py
neuralops/core/triton_kernels/gdn_scan.py
neuralops/core/triton_kernels/gla_scan.py
neuralops/core/triton_kernels/gqa.py
neuralops/core/triton_kernels/hardware_profile.py
neuralops/core/triton_kernels/hgrn2_scan.py
neuralops/core/triton_kernels/linear_attn.py
neuralops/core/triton_kernels/low_rank.py
neuralops/core/triton_kernels/op_registry.py
neuralops/core/triton_kernels/paged_attention.py
neuralops/core/triton_kernels/selective_scan.py
neuralops/core/triton_kernels/sliding_window.py
neuralops/core/triton_kernels/softplus.py
neuralops/core/triton_kernels/sparse_attn.py
neuralops/core/triton_kernels/state_update.py
neuralops/core/triton_kernels/varlen_attention.py
neuralops/core/utils/__init__.py
neuralops/core/utils/config.py
neuralops/core/utils/general.py
neuralops/core/utils/gqa.py
neuralops/core/utils/lora.py
neuralops/core/utils/masks.py
neuralops/core/utils/rope.py
neuralops/data/__init__.py
neuralops/data/augmentation.py
neuralops/data/base.py
neuralops/data/cached_tensors.py
neuralops/data/custom.py
neuralops/data/custom_loader.py
neuralops/data/dataset_discovery.py
neuralops/data/dataset_loader.py
neuralops/data/glue.py
neuralops/data/glue_loader.py
neuralops/data/multi_choice.py
neuralops/data/prepare_datasets.py
neuralops/data/preprocessing.py
neuralops/data/streaming.py
neuralops/data/synthetic_generator.py
neuralops/data/task_mapper.py
neuralops/data/validation.py
neuralops/data/benchmark_datasets/__init__.py
neuralops/data/benchmark_datasets/loaders.py
neuralops/data/benchmark_datasets/manifest.py
neuralops/data/benchmark_datasets/synthetic.py
neuralops/distributed/__init__.py
neuralops/distributed/context_parallel.py
neuralops/distributed/tensor_parallel.py
neuralops/evaluation/__init__.py
neuralops/evaluation/ab_testing.py
neuralops/evaluation/eval_all_models_datasets.py
neuralops/evaluation/eval_data.py
neuralops/evaluation/eval_patcher.py
neuralops/evaluation/eval_perplexity.py
neuralops/evaluation/eval_pipeline.py
neuralops/evaluation/eval_reporter.py
neuralops/evaluation/eval_types.py
neuralops/evaluation/eval_utils.py
neuralops/evaluation/fairness.py
neuralops/evaluation/full_mnli_validation.py
neuralops/evaluation/label_quality.py
neuralops/evaluation/lm_eval_adapter.py
neuralops/evaluation/metrics.py
neuralops/evaluation/niah_passkey.py
neuralops/evaluation/retraining.py
neuralops/evaluation/run_baseline_comparison.py
neuralops/evaluation/run_lm_eval_harness.py
neuralops/experiment_logging/__init__.py
neuralops/experiment_logging/callbacks.py
neuralops/experiment_logging/config.py
neuralops/experiment_logging/integration.py
neuralops/experiment_logging/logger.py
neuralops/experiment_logging/tracker.py
neuralops/experiment_utils/__init__.py
neuralops/experiment_utils/comparison.py
neuralops/experiment_utils/config.py
neuralops/experiment_utils/data.py
neuralops/experiment_utils/main.py
neuralops/experiment_utils/model.py
neuralops/experiment_utils/orchestrator.py
neuralops/experiment_utils/runner.py
neuralops/experiment_utils/tasks.py
neuralops/experiment_utils/train.py
neuralops/export/__init__.py
neuralops/export/exporter.py
neuralops/export/serving.py
neuralops/export/surgery_manifest.py
neuralops/inference/__init__.py
neuralops/inference/engine.py
neuralops/inference/paged_cache.py
neuralops/inference/speculative_decoding.py
neuralops/integrations/__init__.py
neuralops/integrations/distributed_compile.py
neuralops/integrations/hf_automodel.py
neuralops/integrations/serving.py
neuralops/integrations/vllm.py
neuralops/kernels/__init__.py
neuralops/models/__init__.py
neuralops/models/cache.py
neuralops/models/configuration.py
neuralops/models/cuda_graph.py
neuralops/models/gla_hf_model.py
neuralops/models/hub.py
neuralops/models/hybrid_spec.py
neuralops/models/modeling.py
neuralops/models/modeling_block.py
neuralops/models/modeling_mlp.py
neuralops/models/modeling_norm.py
neuralops/models/modeling_rope.py
neuralops/models/paged_state.py
neuralops/models/surgical_io.py
neuralops/results_saver/__init__.py
neuralops/results_saver/config.py
neuralops/results_saver/saver.py
neuralops/results_saver/backends/__init__.py
neuralops/results_saver/backends/huggingface.py
neuralops/results_saver/pipeline/__init__.py
neuralops/results_saver/pipeline/result_writer.py
neuralops/results_saver/storage/__init__.py
neuralops/results_saver/storage/sqlite_ops.py
neuralops/results_saver/writers/__init__.py
neuralops/results_saver/writers/csv_writer.py
neuralops/results_saver/writers/json_parquet_writer.py
neuralops/results_saver/writers/plot_writer.py
neuralops/runners/__init__.py
neuralops/runners/run_master_pipeline.py
neuralops/runners/run_sweep.py
neuralops/runners/unified_orchestrator.py
neuralops/runners/module_sweeps/__init__.py
neuralops/runners/monitor/__init__.py
neuralops/runners/monitor/endpoints.py
neuralops/runners/monitor/rendering.py
neuralops/runners/monitor/server.py
neuralops/runners/monitor/stats.py
neuralops/runners/monitor/telemetry.py
neuralops/surgery/__init__.py
neuralops/surgery/endoskeletal_shell.py
neuralops/surgery/hf_cache_bridge.py
neuralops/surgery/scanner.py
neuralops/surgery/shadow_twin.py
neuralops/surgery/topology_router.py
neuralops/surgery/universal_calibrator.py
neuralops/tracking/__init__.py
neuralops/tracking/backends.py
neuralops/tracking/callbacks.py
neuralops/tracking/tracker.py
neuralops/trainer/__init__.py
neuralops/trainer/base_trainer.py
neuralops/trainer/callbacks.py
neuralops/trainer/custom_loop_trainer.py
neuralops/trainer/foundation_pretrainer.py
neuralops/trainer/hf_callback.py
neuralops/trainer/hf_trainer.py
neuralops/trainer/hyperopt.py
neuralops/trainer/losses.py
neuralops/trainer/metrics.py
neuralops/trainer/registry.py
neuralops/trainer/results_manager.py
neuralops/trainer/rl_trainer.py
neuralops/trainer/schedulers.py
neuralops/trainer/train_lra.py
neuralops/trainer/trainer.py
neuralops/trainer/trainer_factory.py
neuralops/trainer/trainer_svd.py
neuralops/trainer/unsloth_trainer.py
neuralops/trainer/utils.py
neuralops/trainer/distillation/multi_level_kd.py
neuralops/trainer/methods/__init__.py
neuralops/trainer/methods/fisher.py
neuralops/trainer/methods/gelora.py
neuralops/trainer/methods/llrd.py
neuralops/trainer/methods/lora.py
neuralops/trainer/methods/mixout.py
neuralops/trainer/methods/rdrop_r3f.py
neuralops/trainer/methods/registry.py
neuralops/trainer/optimizers/__init__.py
neuralops/trainer/optimizers/factory.py
neuralops/trainer/optimizers/muon.py
neuralops/trainer/optimizers/riemannian.py
neuralops/trainer/optimizers/stiefel.py
neuralops/trainer/peft_recovery/svd_adapters.py
neuralops/trainer/regularizers/attention_stability.py
neuralops/trainer/rl_alignment/preference_and_onpolicy.py
neuralops/trainer/surgery_schedules/mohawk.py
neuralops/trainer/ttt/test_time_trainer.py
neuralops/training/__init__.py
neuralops/training/checkpointing.py
neuralops/training/layer_distill.py
neuralops/utils/__init__.py
neuralops/utils/amp_safety.py
neuralops/utils/architecture.py
neuralops/utils/attention_profiler.py
neuralops/utils/checkpoint.py
neuralops/utils/config.py
neuralops/utils/device.py
neuralops/utils/helpers.py
neuralops/utils/hf_git_exporter.py
neuralops/utils/layer_norm_fix.py
neuralops/utils/logger.py
neuralops/utils/logging_config.py
neuralops/utils/lora.py
neuralops/utils/metrics.py
neuralops/utils/model_family.py
neuralops/utils/paths.py
neuralops/utils/profiler.py
neuralops/utils/qwen_compat.py
neuralops/utils/replace_attention.py
neuralops/utils/results_saver.py
neuralops/utils/seed.py
neuralops/utils/compat/__init__.py
neuralops/utils/compat/aliases.py
neuralops/utils/compat/architectures.py
neuralops/utils/compat/param_recovery.py
neuralops/validation/__init__.py
neuralops/validation/data_validator.py
neuralops/validation/memory_validator.py
neuralops/validation/model_validator.py
neuralops/validation/numerical_validator.py
neuralops/validation/report_generator.py
neuralops/validation/speed_validator.py
neuralops/validation/training_validator.py