AI Systems Architect; Models & Hardware Co-Design

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. You will evaluate emerging model architectures, understand their training and inference characteristics, and help translate them into efficient hardware implementations.

This role requires someone who is comfortable moving between mathematical models, software frameworks, and hardware architecture, and who enjoys solving problems at the boundary between disciplines.

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. You will evaluate emerging model architectures, understand their training and inference characteristics, and help translate them into efficient hardware implementations.

This role requires someone who is comfortable moving between mathematical models, software frameworks, and hardware architecture, and who enjoys solving problems at the boundary between disciplines.

• 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.
• Develop performance and efficiency models to guide architectural decisions.
• Collaborate with software and hardware teams to ensure that models can be deployed efficiently on new compute platforms.
• Stay current with emerging AI model research and identify opportunities where new architectures may benefit from specialized hardware.

• 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 working with ML frameworks such as PyTorch, JAX, or Tensor Flow.
• 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 model optimization techniques such as quantization, sparsity, pruning, or distillation.
• Experience implementing or optimizing AI models for specialized hardware platforms.
• Exposure to emerging AI model architectures beyond transformers (e.g., world models, continuous-time networks, reinforcement learning systems).
• Experience working with robotics, autonomous systems, or embodied AI applications.

$200,000 - $300,000 a year

At Velaura, we believe exceptional talent deserves exceptional rewards. Compensation for this role includes a base salary and very competitive equity compensation, allowing team members to share in the company’s long-term…