Multimodal ML Engineer — Early-Stage, Equity & PTO

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

Benefits offered by this job

Equity

Comprehensive benefits package (health, dental, vision)

Flexible work schedule

PTO

Job summary

A leading AI data partner in California is hiring a Machine Learning Engineer to develop scalable training pipelines and shape the future of multimodal AI systems. The ideal candidate will have a strong background in machine learning, Python proficiency, and experience with modern infrastructure tools. This role offers a competitive salary ranging from $175,000 to $275,000 annually, along with a comprehensive benefits package and equity eligibility.

Qualifications

• Strong academic background in computer science, AI, or ML.
• 3+ years of experience in applied machine learning or ML engineering.
• Proficient in Python and ML frameworks.

Responsibilities

• Design and build scalable ML pipelines for multimodal training.
• Work with data teams to develop efficient data workflows.
• Implement training strategies for large-scale AI systems.

Skills

Applied machine learning experience

Proficient in Python

Experience with PyTorch, TensorFlow, or JAX

ML system design

Excellent communication skills

Education

Master’s degree or Ph.D. in relevant field

Tools

Kubernetes

Ray

Airflow

MLflow

Cloud training environments (AWS, GCP, Azure)

Job description

A leading AI data partner in California is hiring a Machine Learning Engineer to develop scalable training pipelines and shape the future of multimodal AI systems. The ideal candidate will have a strong background in machine learning, Python proficiency, and experience with modern infrastructure tools. This role offers a competitive salary ranging from $175,000 to $275,000 annually, along with a comprehensive benefits package and equity eligibility.