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/serving.py
src/torchbridge/advanced_memory/__init__.py
src/torchbridge/advanced_memory/advanced_checkpointing.py
src/torchbridge/advanced_memory/deep_optimizer_states.py
src/torchbridge/advanced_memory/gradient_compression.py
src/torchbridge/advanced_memory/long_sequence_optimization.py
src/torchbridge/advanced_memory/memory_pool_management.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/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/export.py
src/torchbridge/cli/init.py
src/torchbridge/cli/migrate.py
src/torchbridge/cli/optimize.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/core/optimized_layers/__init__.py
src/torchbridge/core/optimized_layers/activation_functions.py
src/torchbridge/core/optimized_layers/attention_modules.py
src/torchbridge/core/optimized_layers/embedding_layers.py
src/torchbridge/core/optimized_layers/linear_transformations.py
src/torchbridge/core/optimized_layers/normalization_layers.py
src/torchbridge/deployment/__init__.py
src/torchbridge/deployment/export_cli.py
src/torchbridge/deployment/onnx_exporter.py
src/torchbridge/deployment/optimization_metadata.py
src/torchbridge/deployment/production_validator.py
src/torchbridge/deployment/safetensors_exporter.py
src/torchbridge/deployment/torchscript_exporter.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/fsdp2.py
src/torchbridge/distributed/pipeline_schedules.py
src/torchbridge/distributed/topology.py
src/torchbridge/distributed_scale/__init__.py
src/torchbridge/distributed_scale/cluster_management.py
src/torchbridge/distributed_scale/communication_optimization.py
src/torchbridge/distributed_scale/communication_primitives.py
src/torchbridge/distributed_scale/communication_profiling.py
src/torchbridge/distributed_scale/fault_tolerance.py
src/torchbridge/distributed_scale/hardware_adaptation.py
src/torchbridge/distributed_scale/hardware_adapter.py
src/torchbridge/distributed_scale/hardware_discovery.py
src/torchbridge/distributed_scale/job_management.py
src/torchbridge/distributed_scale/large_scale_inference.py
src/torchbridge/distributed_scale/multi_node_training.py
src/torchbridge/distributed_scale/network_optimization.py
src/torchbridge/distributed_scale/orchestration.py
src/torchbridge/distributed_scale/scaling_fault_tolerance.py
src/torchbridge/distributed_scale/thermal_power_management.py
src/torchbridge/distributed_scale/unified_communication.py
src/torchbridge/hardware/__init__.py
src/torchbridge/hardware/abstraction/__init__.py
src/torchbridge/hardware/abstraction/hal_core.py
src/torchbridge/hardware/abstraction/privateuse1_integration.py
src/torchbridge/hardware/abstraction/vendor_adapters.py
src/torchbridge/hardware/gpu/__init__.py
src/torchbridge/hardware/gpu/custom_kernels.py
src/torchbridge/hardware/gpu/memory_optimization.py
src/torchbridge/hardware/gpu/multi_gpu_patterns.py
src/torchbridge/hardware/gpu/profiling_tools.py
src/torchbridge/hardware/gpu/tensor_cores.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/distributed/__init__.py
src/torchbridge/models/distributed/large_model_optimizer.py
src/torchbridge/models/distributed/model_sharding.py
src/torchbridge/models/distributed/pipeline_parallel.py
src/torchbridge/models/distributed/tensor_parallel.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/enhanced_health.py
src/torchbridge/monitoring/grafana_alerts.py
src/torchbridge/monitoring/grafana_dashboards.py
src/torchbridge/monitoring/health_monitor.py
src/torchbridge/monitoring/llm_metrics.py
src/torchbridge/monitoring/prometheus_exporter.py
src/torchbridge/monitoring/slo_framework.py
src/torchbridge/monitoring/structured_logging.py
src/torchbridge/optimizations/__init__.py
src/torchbridge/optimizations/next_gen/__init__.py
src/torchbridge/optimizations/next_gen/pygraph_optimizer.py
src/torchbridge/optimizations/patterns/__init__.py
src/torchbridge/optimizations/patterns/compiler_friendly.py
src/torchbridge/optimizations/patterns/compute_intensity.py
src/torchbridge/optimizations/patterns/dynamic_shapes.py
src/torchbridge/optimizations/patterns/fusion_strategies.py
src/torchbridge/optimizations/patterns/memory_efficiency.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/ab_testing.py
src/torchbridge/utils/cache.py
src/torchbridge/utils/compiler_assistant.py
src/torchbridge/utils/deprecation_manager.py
src/torchbridge/utils/doc_generator.py
src/torchbridge/utils/import_profiler.py
src/torchbridge/utils/model_analyzer.py
src/torchbridge/utils/optimization_recommendations.py
src/torchbridge/utils/profiling.py
src/torchbridge/utils/progressive_optimization.py
src/torchbridge/utils/triton_fused_ops.py
src/torchbridge/utils/universal_inference_engine.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