AI Systems Architect
Analyze modern AI model architectures, evaluate computational and system requirements, and map them to specialized hardware. 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 and workloads.
- Translate model requirements into efficient silicon implementations.
- Optimize models, algorithms, and dataflows for performance, efficiency, and scalability.
- Analyze training and inference pipelines and their 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.
- Strong mathematical foundation in machine learning, optimization, and deep learning algorithms.
- Experience with 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 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