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

Summary

AI Engineer

Data Engineer

Platform Engineer

Founding Engineer

Test Engineer

AI Systems Architect

See also