Advanced AI Infrastructure Design and Optimization
Chapter 1: Chapter 1: Architectural Patterns for AI Platforms
MLOps Principles at Scale
Compute Selection: CPU, GPU, and TPU Architectures
High-Bandwidth Interconnects for Distributed Systems
Storage Solutions for Large-Scale AI Datasets
Networking Topologies for ML Clusters
Hands-on Practical: Environment and Tooling Setup
Chapter 2: Chapter 2: Engineering Distributed Model Training
Data Parallelism with Synchronous and Asynchronous Updates
Model and Pipeline Parallelism for Large Models
Implementing Training with Horovod
Leveraging Microsoft DeepSpeed for ZeRO and Offloading
Fault Tolerance and Checkpointing in Long-Running Jobs
Hands-on Practical: Distributed Training with PyTorch FSDP
Chapter 3: Chapter 3: Advanced Resource Orchestration with Kubernetes
Managing ML Workflows with KubeFlow Pipelines
Advanced GPU Scheduling and Sharing
Cluster Autoscaling for Dynamic ML Workloads
Strategies for Using Spot and Preemptible Instances
Multi-Tenancy with Namespaces, Quotas, and Priority Classes
Practice: Configure a GPU-Aware Autoscaling Group
Chapter 4: Chapter 4: High-Performance Model Inference and Serving
Architecting Inference Services for Latency and Throughput
Model Optimization with TensorRT and ONNX Runtime
Model Quantization Techniques: INT8 and FP8
Serving Multiple Models with NVIDIA Triton Inference Server
A/B Testing and Canary Deployments for Models
Hands-on Practical: Deploying an Optimized Model on Triton
Chapter 5: Chapter 5: Scalable Data Management and Feature Engineering
Designing and Implementing a Feature Store
Real-time vs. Batch Feature Computation
Data Versioning and Lineage with DVC and Pachyderm
High-Throughput Data Processing with Spark and Ray
Managing Data Lakes and Data Warehouses for AI
Practice: Build a Basic Feature Ingestion Pipeline
Chapter 6: Chapter 6: Financial Operations and Governance for AI
Applying FinOps Principles to ML Workloads
Cost Attribution and Showback Models for ML Teams
Optimizing Cloud Storage Costs for Datasets
Right-Sizing Compute for Training and Inference
Automating Cost Anomaly Detection
Governance Policies for Resource Consumption
Practice: Analyzing a Cloud Cost and Usage Report
Hands-on Practical: Distributed Training with PyTorch FSDP
Was this section helpful?
- Fully Sharded Data Parallel (FSDP), PyTorch Team, 2024 - Official documentation for PyTorch's FSDP, detailing its API, usage, and configuration options for distributed training.
- ZeRO: Memory Optimizations Toward Training Trillion Parameter Models, Samyam Rajbhandari, Cong Guo, Jeff Rasley, Shaden Smith, Yuxiong He, 2021 SC21: International Conference for High Performance Computing, Networking, Storage and Analysis (ACM) DOI: 10.1145/3479069.3487532 - This academic paper introduces the ZeRO optimization stages (ZeRO-2 and ZeRO-3) for memory-efficient large model training, which FSDP's sharding strategies are analogous to.
- Distributed communication package - torch.distributed, PyTorch Team, 2024 (PyTorch) - The official PyTorch documentation for its distributed communication primitives, including
init_process_group,torchrun, and environment variables. - Distributed training with 🤗 Accelerate, Hugging Face, 2024 (Hugging Face) - Practical guide from Hugging Face on using Accelerate for simplifying distributed training with transformer models, including FSDP integration.