Abaka AI
Machine Learning Engineer
$175K–$275K
Mountain View HQ · Hybrid · Full-time · Posted Nov 5, 2025
- Experience
- 3+ years in applied machine learning or ML engineering.
- Education
- Strong academic background in computer science, artificial intelligence, machine learning, or a related field; Master’s degree or Ph.D. preferred.
About the job
Build and optimize scalable training pipelines and infrastructure for multimodal AI models, working with data engineering, research, product, and infrastructure teams.
What you'll do
- Design and optimize pipelines for multimodal model training, fine-tuning, and evaluation.
- Develop data workflows and tools for preprocessing, experimentation, tuning, and deployment.
- Improve training and data-pipeline performance, reproducibility, and integration of model outputs.
What they ask for
- 3+ years of experience in applied machine learning or ML engineering, delivering production-ready models or pipelines.
- Proficiency in Python and ML frameworks such as PyTorch, TensorFlow, or JAX, with hands-on large-scale distributed training and inference experience.
- Familiarity with multimodal data processing and dataset optimization for model training.
- Experience with modern infrastructure tools and cloud-based training environments.
Nice to have
- Master’s degree or Ph.D.
Pay
$175,000 - $275,000 USD annually (base salary)
Tools
Python, PyTorch, TensorFlow, JAX, Kubernetes, Ray, Airflow, MLflow
AI focus
Develop and train multimodal AI systems across text, image, audio, video, and 3D data, including vision, video, and diffusion models.
Model-written summary · the original posting is authoritative