LICENSE
README.md
pyproject.toml
setup.py
src/torchbridge/__init__.py
src/torchbridge/py.typed
src/torchbridge/adapters/__init__.py
src/torchbridge/adapters/compatibility.py
src/torchbridge/adapters/config.py
src/torchbridge/adapters/engine.py
src/torchbridge/adapters/layers.py
src/torchbridge/adapters/model_families.py
src/torchbridge/adapters/serving.py
src/torchbridge/attention/__init__.py
src/torchbridge/attention/compatibility/__init__.py
src/torchbridge/attention/core/__init__.py
src/torchbridge/attention/core/attention_ops.py
src/torchbridge/attention/core/base.py
src/torchbridge/attention/core/config.py
src/torchbridge/attention/core/registry.py
src/torchbridge/attention/dispatch/__init__.py
src/torchbridge/attention/dispatch/benchmark_cache.py
src/torchbridge/attention/dispatch/compatibility.py
src/torchbridge/attention/dispatch/dispatcher.py
src/torchbridge/attention/dispatch/kernel_types.py
src/torchbridge/attention/implementations/__init__.py
src/torchbridge/attention/implementations/flash_attention.py
src/torchbridge/attention/implementations/flex_attention.py
src/torchbridge/attention/implementations/memory_efficient.py
src/torchbridge/attention/implementations/sparse.py
src/torchbridge/backends/__init__.py
src/torchbridge/backends/backend_factory.py
src/torchbridge/backends/base_adapter.py
src/torchbridge/backends/base_backend.py
src/torchbridge/backends/base_exceptions.py
src/torchbridge/backends/base_memory_manager.py
src/torchbridge/backends/amd/__init__.py
src/torchbridge/backends/amd/amd_adapter.py
src/torchbridge/backends/amd/amd_backend.py
src/torchbridge/backends/amd/amd_exceptions.py
src/torchbridge/backends/nvidia/__init__.py
src/torchbridge/backends/nvidia/cuda_utilities.py
src/torchbridge/backends/nvidia/flash_attention_integration.py
src/torchbridge/backends/nvidia/fp8_compiler.py
src/torchbridge/backends/nvidia/memory_manager.py
src/torchbridge/backends/nvidia/nvidia_adapter.py
src/torchbridge/backends/nvidia/nvidia_backend.py
src/torchbridge/backends/nvidia/nvidia_exceptions.py
src/torchbridge/backends/tpu/__init__.py
src/torchbridge/backends/tpu/cache_utils.py
src/torchbridge/backends/tpu/memory_manager.py
src/torchbridge/backends/tpu/tpu_adapter.py
src/torchbridge/backends/tpu/tpu_backend.py
src/torchbridge/backends/tpu/tpu_exceptions.py
src/torchbridge/backends/tpu/xla_compat.py
src/torchbridge/backends/tpu/xla_compiler.py
src/torchbridge/backends/tpu/xla_integration.py
src/torchbridge/backends/trainium/__init__.py
src/torchbridge/backends/trainium/memory_manager.py
src/torchbridge/backends/trainium/neuron_compiler.py
src/torchbridge/backends/trainium/neuron_utilities.py
src/torchbridge/backends/trainium/trainium_adapter.py
src/torchbridge/backends/trainium/trainium_backend.py
src/torchbridge/backends/trainium/trainium_exceptions.py
src/torchbridge/benchmarks/__init__.py
src/torchbridge/benchmarks/claim_benchmarks.py
src/torchbridge/benchmarks/claim_registry.py
src/torchbridge/checkpoint/__init__.py
src/torchbridge/checkpoint/config.py
src/torchbridge/checkpoint/frequency.py
src/torchbridge/checkpoint/manager.py
src/torchbridge/checkpoint/metadata.py
src/torchbridge/checkpoint/storage.py
src/torchbridge/cli/__init__.py
src/torchbridge/cli/__main__.py
src/torchbridge/cli/adapter.py
src/torchbridge/cli/advisor.py
src/torchbridge/cli/benchmark.py
src/torchbridge/cli/cache.py
src/torchbridge/cli/checkpoint.py
src/torchbridge/cli/doctor.py
src/torchbridge/cli/init.py
src/torchbridge/cli/migrate.py
src/torchbridge/cli/profile.py
src/torchbridge/cli/quantize.py
src/torchbridge/cli/speculate.py
src/torchbridge/cli/validate.py
src/torchbridge/core/__init__.py
src/torchbridge/core/config.py
src/torchbridge/core/errors.py
src/torchbridge/core/hardware_detector.py
src/torchbridge/core/kernel_registry.py
src/torchbridge/core/performance_tracker.py
src/torchbridge/core/components/__init__.py
src/torchbridge/core/components/basic_optimized.py
src/torchbridge/core/components/jit_optimized.py
src/torchbridge/core/management/__init__.py
src/torchbridge/core/management/base.py
src/torchbridge/core/management/hardware_manager.py
src/torchbridge/core/management/infrastructure_manager.py
src/torchbridge/core/management/optimization_manager.py
src/torchbridge/core/management/unified_manager.py
src/torchbridge/deployment/__init__.py
src/torchbridge/deployment/optimization_metadata.py
src/torchbridge/deployment/production_validator.py
src/torchbridge/deployment/serving/__init__.py
src/torchbridge/deployment/serving/fastapi_server.py
src/torchbridge/deployment/serving/llm_server.py
src/torchbridge/deployment/serving/torchserve_handler.py
src/torchbridge/deployment/serving/triton_config.py
src/torchbridge/distributed/__init__.py
src/torchbridge/distributed/collective_backend.py
src/torchbridge/distributed/config.py
src/torchbridge/distributed/fsdp.py
src/torchbridge/distributed/pipeline_schedules.py
src/torchbridge/distributed/topology.py
src/torchbridge/inference/__init__.py
src/torchbridge/inference/phase_detection.py
src/torchbridge/inference/speculative/__init__.py
src/torchbridge/inference/speculative/compatibility.py
src/torchbridge/inference/speculative/engine.py
src/torchbridge/inference/speculative/methods.py
src/torchbridge/inference/structured/__init__.py
src/torchbridge/inference/structured/output_format.py
src/torchbridge/inference/structured/processor.py
src/torchbridge/mixture_of_experts/__init__.py
src/torchbridge/mixture_of_experts/expert_networks.py
src/torchbridge/mixture_of_experts/moe_layers.py
src/torchbridge/mixture_of_experts/optimization.py
src/torchbridge/mixture_of_experts/routing.py
src/torchbridge/models/__init__.py
src/torchbridge/models/llm/__init__.py
src/torchbridge/models/llm/kv_cache.py
src/torchbridge/models/llm/llm_optimizer.py
src/torchbridge/models/llm/kv/__init__.py
src/torchbridge/models/llm/kv/cache_compatibility.py
src/torchbridge/models/llm/kv/cache_dtype.py
src/torchbridge/models/llm/kv/quantized_cache.py
src/torchbridge/monitoring/__init__.py
src/torchbridge/monitoring/llm_metrics.py
src/torchbridge/precision/__init__.py
src/torchbridge/precision/fp4_native.py
src/torchbridge/precision/fp8_native.py
src/torchbridge/precision/fp8_training_engine.py
src/torchbridge/precision/quantization/__init__.py
src/torchbridge/precision/quantization/compatibility.py
src/torchbridge/precision/quantization/engine.py
src/torchbridge/precision/quantization/formats.py
src/torchbridge/precision/quantization/torchao_integration.py
src/torchbridge/testing/__init__.py
src/torchbridge/testing/decorator.py
src/torchbridge/testing/divergence.py
src/torchbridge/testing/plugin.py
src/torchbridge/testing/report.py
src/torchbridge/testing/suite.py
src/torchbridge/testing/tolerance_db.py
src/torchbridge/utils/__init__.py
src/torchbridge/utils/cache.py
src/torchbridge/utils/deprecation_manager.py
src/torchbridge/utils/model_analyzer.py
src/torchbridge/utils/profiling.py
src/torchbridge/validation/__init__.py
src/torchbridge/validation/unified_validator.py
src/torchbridge_ml.egg-info/PKG-INFO
src/torchbridge_ml.egg-info/SOURCES.txt
src/torchbridge_ml.egg-info/dependency_links.txt
src/torchbridge_ml.egg-info/entry_points.txt
src/torchbridge_ml.egg-info/requires.txt
src/torchbridge_ml.egg-info/top_level.txt