Abaka AI ยท 5 hours ago
Machine Learning Engineer
Palo Alto, CA
Full-time
Onsite
Mid, Senior Level
$175K/yr - $275K/yr
3+ years expAbaka AI is a trusted data partner for AI companies, specializing in scalable data solutions. They are seeking a Machine Learning Engineer to design and develop training pipelines, collaborate with engineering teams, and influence model development strategies.
Data Collection and LabelingMachine LearningNatural Language Processing
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
Qualification
Required
Strong academic background in computer science, artificial intelligence, machine learning, or related fields
3+ years of experience in applied machine learning or ML engineering, with a demonstrated ability to deliver production-ready models or pipelines
Proficient in Python and ML frameworks such as PyTorch, TensorFlow, or JAX, with hands-on experience in large-scale distributed training and inference systems
Familiarity with multimodal data processing (e.g., text-image pairing, video understanding, speech-audio modeling) and dataset optimization for model training
Solid understanding of ML system design, including feature pipelines, data loaders, model serving, and evaluation frameworks
Experience with modern infrastructure tools such as Kubernetes, Ray, Airflow, or MLflow, along with cloud-based training environments (AWS, GCP, Azure)
Excellent communication and collaboration skills, capable of working effectively across engineering, research, and product teams to accomplish shared goals
Self-driven and adaptable, comfortable operating in a fast-paced startup environment, and able to demonstrate strong ownership and urgency in execution
Preferred
Master's degree or Ph.D. is preferred
Benefits
Equity
Comprehensive benefits package
Health
Dental
Vision
PTO
Flexible work schedule
Company
Abaka AI
Abaka AI is a leading AI company and we are committed to becoming the data partner in artificial intelligence industry.
51-200 employees
H1B Sponsorship
Abaka AI has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (2)
Funding
Current Stage
Growth StageLeadership Team
Minyi (Crystal) Chen
North American Business Partner
Company data provided by crunchbase