Machine Learning Engineer

Abaka AI is a data partner for AI companies, supporting over 1,000 industry leaders with scalable data solutions. They are seeking a Machine Learning Engineer to design and optimize training pipelines for multimodal AI systems, collaborating closely with data engineering and research 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