ML Engineer – Multimodal AI Pipelines & Training

Abaka AI; Mountain View (CA); On-site; USD 175,000 - 275,000; Full time

Benefits offered by this job

Equity

Health, dental, vision, PTO, flexible

Job summary

Abaka AI in Silicon Valley is hiring a Machine Learning Engineer to shape how we build, train, and optimize multimodal AI systems. You will own scalable training pipelines and work with data engineering and research teams to drive model development across text, image, audio, video, and 3D data.

As an early member of the engineering team, you will influence model training strategy, experimentation frameworks, distributed infrastructure, and governance practices to ensure reproducibility and high

Qualifications

• 3+ years of experience in applied machine learning or ML engineering.
• Proficient in Python and ML frameworks such as PyTorch, TensorFlow or JAX.
• Experience with multimodal data processing and dataset optimization for model training.
• Knowledge of ML system design: data loaders, feature pipelines, model serving, evaluation.
• Experience with Kubernetes, Ray, Airflow or MLflow and cloud training environments (AWS, GCP, Azure).
• Strong communication and collaboration across engineering, research and product teams.

Responsibilities

• Design, build, and optimize scalable machine learning pipelines for multimodal model training, fine-tuning, and evaluation across text, image, audio, video, and 3D data.
• Work closely with data engineering and research teams to develop efficient data workflows, including collection, preprocessing, annotation, versioning, and model integration.
• Implement and refine training strategies for large-scale AI systems, including vision, video, and diffusion models, ensuring reproducibility, efficiency, and strong model performance.
• Develop tools and automation frameworks that accelerate model experimentation, hyperparameter tuning, and deployment.
• Identify and address performance bottlenecks in data or training pipelines to improve throughput, stability, and resource utilization.
• Collaborate with product and infrastructure teams to ensure smooth integration of model outputs into both internal and client-facing applications.
• Support internal best practices for model governance, experiment tracking, and documentation to maintain high engineering standards and reproducibility.

Skills

Python

PyTorch

TensorFlow

JAX

Distributed training

Multimodal data

Kubernetes

Ray

Airflow

Education

Master's or PhD preferred

Tools

Kubernetes

Ray

Airflow

MLflow

AWS/GCP/Azure

Job description

Abaka AI in Silicon Valley is hiring a Machine Learning Engineer to shape how we build, train, and optimize multimodal AI systems. You will own scalable training pipelines and work with data engineering and research teams to drive model development across text, image, audio, video, and 3D data.

As an early member of the engineering team, you will influence model training strategy, experimentation frameworks, distributed infrastructure, and governance practices to ensure reproducibility and high