AI Systems Architect: Models & Hardware Equity & Flex

Velaura; Boston (MA); On-site; USD 200,000 - 500,000; Full time

Benefits offered by this job

Medical, dental, and vision coverage

Paid time off

Flexible work arrangements

Professional development opportunities

Equity participation

Job summary

Velaura is seeking an AI Systems Architect in Boston to bridge machine learning models and hardware design. You will shape AI workloads, evaluate model architectures, and translate requirements into efficient silicon implementations.

You will collaborate across software, algorithms, and hardware teams to deploy models on new compute platforms and stay at the forefront of AI research and hardware co-design.

Qualifications

• Strong understanding of modern ML architectures, including transformers.
• Solid grounding in the mathematical foundations of machine learning, optimization, and deep learning algorithms.
• Experience working 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.

Responsibilities

• 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.

Skills

Transformer architectures

Math foundations

ML frameworks (PyTorch, JAX, TF)

Dataflow analysis

Model training/inference workflows

Systems thinking across software/hw

Tools

PyTorch

JAX

TensorFlow

Job description

Velaura is seeking an AI Systems Architect in Boston to bridge machine learning models and hardware design. You will shape AI workloads, evaluate model architectures, and translate requirements into efficient silicon implementations.

You will collaborate across software, algorithms, and hardware teams to deploy models on new compute platforms and stay at the forefront of AI research and hardware co-design.