AI Systems Architect
Summary
Designs and optimizes AI model architectures for specialized hardware, mapping model requirements to efficient silicon implementations and improving performance and scalability.
You analyze modern AI model architectures, evaluate their computational and system requirements, and determine how they map to specialized hardware. You translate model requirements into efficient silicon implementations, optimize models and dataflows, develop performance and efficiency models, and collaborate across software, algorithms, and hardware disciplines.
Responsibilities
- Analyze modern AI model architectures including transformers and emerging alternatives
- Evaluate model suitability for different application domains
- Translate model requirements into efficient silicon implementations
- Identify opportunities to optimize models, algorithms, and dataflows for performance, efficiency, and scalability
- Understand training and inference pipelines and identify implications for hardware design
- Develop performance and efficiency models to guide architectural decisions
- Collaborate with software and hardware teams to deploy models efficiently on new compute platforms
- Stay current with emerging AI model research
Requirements
- Strong understanding of modern machine learning architectures, including transformers and related model families
- Solid grounding in the mathematical foundations of machine learning, optimization, and deep learning algorithms
- Experience with ML frameworks such as PyTorch, JAX, or TensorFlow
- Ability to analyze model computation graphs and translate them into efficient dataflows and compute patterns
- Experience with AI model training and inference workflows
- Strong systems thinking and ability to work across software, algorithms, and hardware
- Experience with hardware-software co-design or AI accelerator architecture
- Familiarity with quantization, sparsity, pruning, or distillation
- Experience implementing or optimizing AI models for specialized hardware platforms
- Exposure to emerging AI model architectures beyond transformers
- Experience with robotics, autonomous systems, or embodied AI applications
Benefits
- Performance-based incentives
- Equity participation
- Medical coverage
- Dental coverage
- Vision coverage
- Paid time off
- Flexible work arrangements
- Professional development opportunities