AI Systems Architect (Models & Hardware Co-Design) - Bangalore

About the role

We are looking for an AI Systems Architect who sits at the intersection of machine learning models and hardware architecture. In this role, you will work across model development, algorithms, and hardware architecture to identify and shape the AI workloads that will define the next generation of compute platforms.

What you'll do

• Analyze modern AI model architectures including transformers and emerging alternatives to understand their computational and system requirements
• Evaluate model suitability for different application domains and help determine which model architectures are best matched to specific workloads
• Work closely with hardware architects to translate model requirements into efficient silicon implementations
• Identify opportunities to modify or optimize models, algorithms, or dataflows to improve performance, efficiency, or scalability in hardware
• Understand training and inference pipelines for modern AI models and identify implications for hardware design

What you need

• Hardware
• Machine learning
• Deep learning
• PyTorch
• TensorFlow
• Any graduate