AI/ML Engineer – (Next-Generation AI Platforms & Workloads)

VeeAR Projects Inc.; Sunnyvale (CA); On-site; USD 140,000 - 210,000; Full time

Job summary

A progressive AI engineering firm based in Sunnyvale, California, is seeking an AI Engineer to design and optimize cutting-edge AI models. You will leverage high-performance GPU clusters and collaborate with cross-functional teams to enhance performance and scalability. The ideal candidate will have a Bachelor's in a related field and 3+ years of practical experience in machine learning, deep learning, and software engineering. This role offers competitive benefits and opportunities for growth within a dynamic environment.

Qualifications

• 3+ years of hands-on experience in machine learning and software engineering.
• Strong working knowledge of AI/ML frameworks.
• Familiarity with distributed training frameworks is a plus.

Responsibilities

• Design and train state-of-the-art ML models for various applications.
• Optimize AI workloads for large-scale GPU systems.
• Collaborate with teams to maximize AI processing efficiency.
• Build robust tools, data pipelines, evaluation frameworks, and deployment systems.
• Track and incorporate the latest AI research and technological advancements.
• Contribute to product requirements and agile execution.

Skills

Machine learning

Deep learning

Python

C/C++

Data structures

Algorithms

Software design principles

Generative AI

Computer Vision

Ray

DeepSpeed

Megatron-LM

Kubeflow

MLflow

Education

Bachelor's degree in Computer Science or related field

Master's or PhD in Computer Science, AI/ML, or a related discipline

Tools

PyTorch

TensorFlow

JAX

Dynamo

vLLM

TensorRT

Triton Inference Server

Job description

Role Overview

As a core member of our AI engineering team, you will design, develop, and optimize cutting‑edge AI models and workloads that run natively on our high‑performance GPU clusters. Leverage our SOTA infrastructure to train, fine‑tune, and serve massive‑scale models at unprecedented efficiency. Collaborate across infrastructure, product, and research teams to align hardware capabilities with real‑world AI demands, driving breakthroughs in performance, scalability, and innovation.

Key Responsibilities

• Design, implement, and train state‑of‑the‑art ML models for high‑impact applications (e.g., NLP, Computer Vision, Network Optimization).
• Optimize AI workloads for extreme performance and scalability on large‑scale GPU systems like GB200 NVL72, using tools such as Dynamo, vLLM, and advanced inference engines.
• Partner with cross‑functional teams to co‑design hardware‑software solutions that maximize AI processing efficiency.
• Build robust tools, data pipelines, evaluation frameworks, and deployment systems.
• Track and incorporate the latest AI research and technological advancements.
• Contribute to product requirements (PRDs) and agile execution (sprint planning and delivery).
• Champion a culture of humility, bold innovation, and high‑velocity product delivery.

Minimum Qualifications

• Bachelor's degree in Computer Science, Electrical Engineering, Mathematics, Statistics, or a related technical field.
• 3+ years of hands‑on experience in machine learning, deep learning, and software engineering.
• Proficiency in Python; experience with C/C++.
• Strong working knowledge of major AI/ML frameworks (PyTorch, TensorFlow, JAX, or similar).
• Solid foundation in data structures, algorithms, and software design principles.

Preferred Qualifications

• Master's or PhD in Computer Science, AI/ML, or a related discipline.
• Experience with Large Language Models (LLMs), Generative AI, or Computer Vision.
• Familiarity with distributed training frameworks and techniques (e.g., Ray, DeepSpeed, Megatron‑LM).
• Proven expertise optimizing models for GPU inference (e.g., TensorRT, Triton Inference Server).
• Knowledge of MLOps tools and practices (Kubeflow, MLflow, etc.).