ML Engineer

About The Company

Yobi is a rapidly growing Behavioral AI company dedicated to ethically democratizing the benefits of data and artificial intelligence. Since its inception in 2019, Yobi has established itself as a leader in building one of the largest consented behavioral datasets in the United States, extending beyond the confines of traditional Big Tech ecosystems. Unlike conventional large language model providers, Yobi focuses on constructing foundational models of human behavior grounded in real-world actions such as purchases and store visits. This approach enables highly personalized and decisioning capabilities while maintaining a strong commitment to privacy, safety, and ethics.

The company's innovative private-by-design modeling allows for state-of-the-art personalization for leading brands and agencies, empowering them to achieve walled-garden results without the associated restrictions. With a vision to bring the performance of closed-web user acquisition to the open web and connected TV, Yobi is building the behavioral intelligence layer for any system making personalization decisions. The company is well-funded, with strategic partnerships with industry giants like Microsoft and Databricks, and boasts a world-class team of machine learning experts with backgrounds from Amazon, Uber, Twitter, and Meta. Yobi offers a dynamic, innovative work environment with opportunities for growth and impact, whether working remotely or from multiple hubs across the United States.

About The Role

As a Machine Learning Engineer specializing in Inference and Serving at Yobi, you will be instrumental in designing, optimizing, and operating the systems that deliver our Behavioral AI models in real-time. Your work will be at the heart of our production environment, transforming trained models into high-performance, reliable services that power our open-web and connected TV products. This role is highly applied, blending engineering expertise, deployment strategies, and model understanding to ensure every prediction is fast, accurate, and accountable.

You will be responsible for packaging, versioning, rolling out, and monitoring models across various environments, ensuring seamless and efficient inference processes. Success in this role requires a deep understanding of model deployment, low-latency inference techniques, robust system design, and operational maturity. You will collaborate closely with research teams to interpret model performance, work on optimizing deployment pipelines, and ensure the scalability and reproducibility of serving systems. This position offers an exciting opportunity to shape the future of Behavioral AI systems and contribute to products that impact a broad spectrum of digital advertising and personalization platforms.

Qualifications

• Bachelor's or Master's degree in Computer Science, Engineering, or a related field
• Proven experience in deploying and scaling machine learning models in production environments
• Strong programming skills in Go, Rust, C++, Java, or similar languages
• Experience with Python for model integration and analysis
• Deep understanding of low-latency inference techniques such as model graph optimization, quantization, caching, and batching
• Familiarity with model versioning, rollouts, and live experimentation strategies
• Operational experience with monitoring, drift detection, and observability of ML systems
• Knowledge of infrastructure design for reproducibility and portability, including model registries and lightweight orchestration
• Ability to reason about model performance, interpret trade-offs, and collaborate with research teams
• Strong problem-solving skills, attention to detail, and a proactive attitude

Responsibilities

• Design, develop, and optimize real-time inference systems for Behavioral AI models
• Implement scalable and low-latency deployment pipelines for machine learning models
• Manage model versioning, rollouts, and rollback strategies to ensure stability and performance
• Monitor inference systems for drift, performance issues, and operational health
• Collaborate with research teams to improve model deployability and interpretability
• Ensure reproducibility and portability of serving systems through effective infrastructure design
• Implement caching, batching, and feature retrieval optimizations to enhance inference speed
• Maintain high standards of system reliability, security, and compliance with privacy policies
• Contribute to the development of best practices for ML deployment and operational excellence

Benefits

• Competitive base salary aligned with industry standards
• Meaningful equity and potential financial upside as a percentage of the company
• Annual bonus target based on individual and company performance
• Comprehensive health, dental, and vision insurance plans with minimal out-of-pocket costs
• Unlimited paid time off to promote work-life balance and impact-driven work
• 401(k) plan with company matching contributions
• Fully remote work options or hybrid arrangements from several U.S. hubs (San Francisco Bay Area, Seattle, New York City)
• Opportunity to work with a world-class team of ML experts and industry veterans
• Engagement in innovative projects shaping the future of Behavioral AI and personalization systems

Equal Opportunity

Yobi is an equal opportunity employer committed to fostering an inclusive environment for all employees. We celebrate diversity and are dedicated to creating a workplace where everyone feels valued, respected, and empowered to contribute their best. We do not discriminate based on race, color, religion, gender, gender identity or expression, sexual orientation, national origin, age, disability, or any other protected characteristic. All qualified applicants will receive consideration for employment without regard to these factors.