.gitignore
README.md
pyproject.toml
.github/CODEOWNERS
.github/FUNDING.yml
.github/dependabot.yml
.github/ISSUE_TEMPLATE/bug---issue.md
.github/ISSUE_TEMPLATE/feature-request.md
.github/workflows/consolidated-tests-ci.yml
.github/workflows/lint-ci.yml
.github/workflows/mlx-ci.yml
.github/workflows/security-audit.yml
.github/workflows/stale.yml
.github/workflows/studio-export-fix-ci.yml
.github/workflows/wheel-smoke.yml
scripts/enforce_kwargs_spacing.py
scripts/lint_workflow_triggers.py
scripts/scan_packages.py
scripts/verify_comment_only_diff.py
tests/_merge_e2e_helpers.py
tests/_pickle_base.py
tests/conftest.py
tests/test_active_merge_device_matrix.py
tests/test_backend_device_helpers.py
tests/test_compiler_dynamic_exec.py
tests/test_compiler_output_capture.py
tests/test_compiler_rewriter_exhaustive.py
tests/test_convert_hf_to_gguf_patcher.py
tests/test_convert_to_gguf_self_heal.py
tests/test_deepseek_v2_moe_alias.py
tests/test_diffusion_canvas_maxtok.py
tests/test_diffusion_shim_thought_split.py
tests/test_diffusion_visual_engine_env.py
tests/test_encode_conversations_with_harmony.py
tests/test_encode_conversations_with_harmony_render.py
tests/test_extended_dep_api_pins.py
tests/test_file_uri_resolution.py
tests/test_forward_native_moe_loop_lora.py
tests/test_fused_forward_install.py
tests/test_gemma4_clippable_linear_peft_reload.py
tests/test_gemma4_moe_lora_registration.py
tests/test_gpt_oss_attention_mask.py
tests/test_hf_cache_redirect.py
tests/test_llama_cpp_bundle_converter.py
tests/test_llama_cpp_loader.py
tests/test_llama_cpp_prebuilt.py
tests/test_merge_e2e_dense.py
tests/test_merge_e2e_moe_fused.py
tests/test_merge_e2e_moe_per_expert.py
tests/test_merge_e2e_resized.py
tests/test_merge_e2e_vision_passthrough.py
tests/test_mlx_adapter_metadata_persistence.py
tests/test_mlx_baseline_loss_parity.py
tests/test_mlx_batch_padding.py
tests/test_mlx_batching_and_decay.py
tests/test_mlx_cce_kernel.py
tests/test_mlx_cce_target_classification.py
tests/test_mlx_dequantize_modules.py
tests/test_mlx_dtype_downcast_warning.py
tests/test_mlx_finetune_last_n_layers.py
tests/test_mlx_gated_delta.py
tests/test_mlx_gated_delta_batch_grad.py
tests/test_mlx_get_peft_model_seed_ordering.py
tests/test_mlx_grad_clip_resolution.py
tests/test_mlx_module_exports.py
tests/test_mlx_pr684_full_validation_metal.py
tests/test_mlx_runtime_cce_compile.py
tests/test_mlx_save_export_edge_cases.py
tests/test_mlx_save_export_regressions.py
tests/test_mlx_save_lora_adapters_filter.py
tests/test_mlx_torch_shim_smoke.py
tests/test_mlx_trainer_internals.py
tests/test_mlx_training_e2e_metal.py
tests/test_mlx_vlm_label_masks.py
tests/test_moe_fused_narrow_expert_merge.py
tests/test_moe_grouped_mm_alignment_fallback.py
tests/test_moe_grouped_mm_format.py
tests/test_moe_lora_extractor_coverage.py
tests/test_moe_merge_e2e_cpu.py
tests/test_moe_num_experts_header_precedence.py
tests/test_moe_quant_cleanup.py
tests/test_moe_quant_handler_registry.py
tests/test_offline_env_cross_sync.py
tests/test_patch_loss_functions_coverage.py
tests/test_peft_paramwrapper_layout_drift.py
tests/test_pr684_review_fixes_a.py
tests/test_pr684_review_fixes_b.py
tests/test_prepare_model_for_training_fp32_norms.py
tests/test_pypi_version_sync.py
tests/test_quantize_gguf_q2_k_l.py
tests/test_qwen35_ssm_tensor_mapping.py
tests/test_qwen35_vjp_metal.py
tests/test_qwen_moe_lora_extractor.py
tests/test_resized_shard_rewrite_parity.py
tests/test_rl_replacements_cpu.py
tests/test_saving_utils_lora_remap_count.py
tests/test_saving_utils_quant_aware_merge.py
tests/test_temporary_patches_exhaustive.py
tests/test_temporary_patches_imports.py
tests/test_training_utils_use_cache.py
tests/test_transformers_moe_structure_drift.py
tests/test_unsloth_zoo_lora_merge.py
tests/test_upstream_import_fixes_drift.py
tests/test_upstream_pinned_symbols_accelerator.py
tests/test_upstream_pinned_symbols_transformers.py
tests/test_upstream_pinned_symbols_trl_vllm.py
tests/test_upstream_signatures.py
tests/test_upstream_source_patterns.py
tests/test_vision_collator_audio.py
tests/test_vision_collator_prompt_completion.py
tests/test_vllm_sleep_cache_reset.py
tests/test_vllm_to_hf_conversion.py
tests/test_zoo_history_regressions.py
tests/test_zoo_history_regressions_deep.py
tests/test_zoo_source_upstream_refs.py
tests/mlx_simulation/__init__.py
tests/mlx_simulation/mlx_lm_stub.py
tests/mlx_simulation/mlx_nn_stub.py
tests/mlx_simulation/mlx_optimizers_stub.py
tests/mlx_simulation/mlx_stub.py
tests/mlx_simulation/mlx_utils_stub.py
tests/mlx_simulation/mlx_vlm_stub.py
tests/mlx_simulation/mlx_helpers/__init__.py
tests/mlx_simulation/mlx_helpers/array_proxy.py
tests/mlx_simulation/mlx_helpers/custom_function.py
tests/mlx_simulation/mlx_helpers/lora_linear.py
tests/mlx_simulation/mlx_helpers/mlx_lm_compat.py
tests/mlx_simulation/mlx_helpers/quant.py
tests/mlx_simulation/mlx_helpers/stream_generate.py
tests/mlx_simulation/mlx_helpers/value_and_grad.py
tests/security/__init__.py
tests/security/conftest.py
tests/security/test_lint_workflow_triggers.py
tests/security/test_scan_packages.py
tests/security/fixtures/__init__.py
tests/security/fixtures/_build.py
tests/security/fixtures/clean_wheel.whl
tests/security/fixtures/malicious_sdist.tar.gz
tests/security/fixtures/malicious_wheel.whl
tunelabs_zoo/__init__.py
tunelabs_zoo/compiler.py
tunelabs_zoo/compiler_replacements.py
tunelabs_zoo/dataset_utils.py
tunelabs_zoo/device_type.py
tunelabs_zoo/empty_model.py
tunelabs_zoo/gated_delta_vjp.py
tunelabs_zoo/gradient_checkpointing.py
tunelabs_zoo/hf_cache.py
tunelabs_zoo/hf_utils.py
tunelabs_zoo/llama_cpp.py
tunelabs_zoo/log.py
tunelabs_zoo/logging_utils.py
tunelabs_zoo/loss_utils.py
tunelabs_zoo/model_lists.py
tunelabs_zoo/patch_torch_functions.py
tunelabs_zoo/patching_utils.py
tunelabs_zoo/peft_utils.py
tunelabs_zoo/rl_environments.py
tunelabs_zoo/rl_replacements.py
tunelabs_zoo/saving_utils.py
tunelabs_zoo/tiled_mlp.py
tunelabs_zoo/tokenizer_utils.py
tunelabs_zoo/training_utils.py
tunelabs_zoo/utils.py
tunelabs_zoo/vision_utils.py
tunelabs_zoo/vllm_lora_request.py
tunelabs_zoo/vllm_lora_worker_manager.py
tunelabs_zoo/vllm_rlhf_utils.py
tunelabs_zoo/vllm_utils.py
tunelabs_zoo.egg-info/PKG-INFO
tunelabs_zoo.egg-info/SOURCES.txt
tunelabs_zoo.egg-info/dependency_links.txt
tunelabs_zoo.egg-info/requires.txt
tunelabs_zoo.egg-info/top_level.txt
tunelabs_zoo/diffusion_studio/__init__.py
tunelabs_zoo/diffusion_studio/canvas_player.html
tunelabs_zoo/diffusion_studio/shim.py
tunelabs_zoo/diffusion_studio/visual_engine.py
tunelabs_zoo/flex_attention/__init__.py
tunelabs_zoo/flex_attention/attention_sink.py
tunelabs_zoo/flex_attention/utils.py
tunelabs_zoo/fused_losses/__init__.py
tunelabs_zoo/fused_losses/ast_rewriter.py
tunelabs_zoo/fused_losses/cross_entropy_loss.py
tunelabs_zoo/fused_losses/forward_adapter.py
tunelabs_zoo/fused_losses/forward_install.py
tunelabs_zoo/fused_losses/tests.py
tunelabs_zoo/mlx/__init__.py
tunelabs_zoo/mlx/compile.py
tunelabs_zoo/mlx/loader.py
tunelabs_zoo/mlx/runtime.py
tunelabs_zoo/mlx/trainer.py
tunelabs_zoo/mlx/utils.py
tunelabs_zoo/mlx/cce/__init__.py
tunelabs_zoo/mlx/cce/runtime_cce.py
tunelabs_zoo/stubs/__init__.py
tunelabs_zoo/stubs/bitsandbytes_stub.py
tunelabs_zoo/stubs/triton_stub.py
tunelabs_zoo/temporary_patches/__init__.py
tunelabs_zoo/temporary_patches/bitsandbytes.py
tunelabs_zoo/temporary_patches/common.py
tunelabs_zoo/temporary_patches/deepseek_v3_moe.py
tunelabs_zoo/temporary_patches/ernie4_5_moe.py
tunelabs_zoo/temporary_patches/flex_attention_bwd.py
tunelabs_zoo/temporary_patches/gemma.py
tunelabs_zoo/temporary_patches/gemma3n.py
tunelabs_zoo/temporary_patches/gemma4.py
tunelabs_zoo/temporary_patches/gemma4_moe.py
tunelabs_zoo/temporary_patches/glm4_moe.py
tunelabs_zoo/temporary_patches/gpt_oss.py
tunelabs_zoo/temporary_patches/lfm2_moe.py
tunelabs_zoo/temporary_patches/ministral.py
tunelabs_zoo/temporary_patches/misc.py
tunelabs_zoo/temporary_patches/moe_bnb.py
tunelabs_zoo/temporary_patches/moe_utils.py
tunelabs_zoo/temporary_patches/moe_utils_bnb4bit.py
tunelabs_zoo/temporary_patches/moe_utils_fp8.py
tunelabs_zoo/temporary_patches/mxfp4.py
tunelabs_zoo/temporary_patches/pixtral.py
tunelabs_zoo/temporary_patches/qwen3_5_moe.py
tunelabs_zoo/temporary_patches/qwen3_moe.py
tunelabs_zoo/temporary_patches/qwen3_next_moe.py
tunelabs_zoo/temporary_patches/qwen3_vl_moe.py
tunelabs_zoo/temporary_patches/utils.py