Yobi AI Logo

Yobi AI

Machine Learning Engineer - Inference / Serving

Reposted 7 Days Ago
In-Office or Remote
2 Locations
Mid level
In-Office or Remote
2 Locations
Mid level
As a Machine Learning Engineer at Yobi, you will design and optimize systems for real-time model serving and inference, ensuring performance and reliability.
The summary above was generated by AI

Yobi is a rapidly growing Behavioral AI company on a mission to ethically democratize the benefits of data and AI.

Since 2019, we have built one of the largest consented behavioral datasets in the United States, extending far beyond the walled gardens of Big Tech. Unlike traditional LLM companies, Yobi builds foundation models of human behavior grounded in real-world actions such as purchases and store visits.

Our private-by-design modeling enables state-of-the-art personalization and decisioning for leading brands and agencies while protecting privacy, safety, and ethics.

Today, we are focused on bringing the performance of closed-web user acquisition to the open web and connected TV, giving brands walled-garden results without the walls.

At our core, Yobi is building the behavioral intelligence layer for any system that makes a personalization decision.

Working at Yobi

We’re at an inflection point—customer adoption is accelerating, but there’s still room to shape the architecture and culture from the ground up. Engineers here own major surface areas, build 0→1 systems in large-scale data and model infrastructure, and help define how Behavioral AI scales ethically and effectively.

Highlights:

  • Well-funded with 5+ years of runway. At the same time, we are scaling revenue quickly and project to be breakeven in 2026.

  • Partnerships with Microsoft and Databricks

  • Fully remote or hybrid from several hubs (SF Bay Area, Seattle, NYC)

  • World-class team of Machine Learning experts who worked on cutting edge infra and recommender systems @ Amazon, Uber, Twitter, Meta, etc.

  • Product and Go-To-Market teams who have taken ideas from concept to 9 figure revenue streams

Benefits:

  • Competitive Base Salary

  • Meaningful equity & financial upside - a real % of the company

  • Annual bonus target based on personal and company performance

  • Health, Dental, Vision - most plans will pay little to 0 out of pocket

  • Unlimited PTO - we care about impact, not tracking days you’re out

  • 401k with company match %

About The Role

As a Machine Learning Engineer focused on Inference and Serving at Yobi, you’ll design, optimize, and operate the systems that bring our Behavioral AI models to life in real time. You’ll work at the core of our production environment, turning trained models into performant, reliable, and continuously improving services that power our open-web and CTV products.

This is an applied ML systems role—equal parts engineering depth, deployment craft, and model intuition. You’ll shape how models are packaged, versioned, rolled out, and observed across environments, ensuring every prediction is fast, accurate, and accountable.

What it takes to succeed in this role:
  • Deep expertise in model deployment. You’ve built or scaled production ML serving systems—handling versioning, rollouts, rollback strategies, and live experimentation.

  • Low-latency mindset. You understand what makes inference fast: model graph optimization, quantization, caching, batching, and efficient feature retrieval.

  • Systems fluency. You write robust, high-performance code in Go, Rust, C++, or Java, and are comfortable bridging to Python for model integration and analysis.

  • Operational maturity. You treat inference as a living system—monitoring drift, tracking model lineage, and ensuring observability from input to outcome.

  • Infrastructure intuition. You know how to make serving systems reproducible and portable without over-engineering them, whether that’s through custom runtime design, model registries, or lightweight orchestration.

  • Applied ML understanding. You can reason about model performance, interpret trade-offs, and work with researchers to make models more deployable.

We prioritize attitude, culture, and general (technical) fit over matching perfectly into one of our job descriptions. If our mission and work resonates with you, we encourage you to apply. Tell us how you can help drive our products forward, even if you don’t feel like you are a perfect fit for some of the listings. 

Top Skills

C++
Go
Java
Python
Rust

Similar Jobs

52 Minutes Ago
Remote or Hybrid
Colorado, USA
172K-193K Annually
Expert/Leader
172K-193K Annually
Expert/Leader
eCommerce • Mobile • Payments
Lead the Client Analytics team, develop analytics talent, deliver insights for clients, and drive Client Analytics strategy. Manage relationships and projects.
Top Skills: Business AnalyticsData ScienceStatistics
56 Minutes Ago
Remote or Hybrid
Los Angeles, CA, USA
102K-128K Annually
Senior level
102K-128K Annually
Senior level
Cloud • Fintech • Information Technology • Machine Learning • Software • App development • Generative AI
The Senior PS Solutions Consultant sells services related to SaaS accounting software, manages sales processes, estimates implementation efforts, and prepares Statements of Work while guiding customers through the adoption process.
Top Skills: Accounting SoftwareCrm ToolsProject Management SoftwareSaaS
An Hour Ago
Remote or Hybrid
4 Locations
170K-230K Annually
Senior level
170K-230K Annually
Senior level
Artificial Intelligence • Big Data • Healthtech • Machine Learning • Analytics • Biotech • Generative AI
The Staff Machine Learning Engineer will architect and optimize data infrastructure for multimodal AI models in oncology, focusing on data management and processing pipelines.
Top Skills: AirflowSparkDaskGCPHugging Face DatasetsKubeflowKubeflow PipelinesMlflowMosaicml StreamingPythonRaySagemaker Pipelines

What you need to know about the Austin Tech Scene

Austin has a diverse and thriving tech ecosystem thanks to home-grown companies like Dell and major campuses for IBM, AMD and Apple. The state’s flagship university, the University of Texas at Austin, is known for its engineering school, and the city is known for its annual South by Southwest tech and media conference. Austin’s tech scene spans many verticals, but it’s particularly known for hardware, including semiconductors, as well as AI, biotechnology and cloud computing. And its food and music scene, low taxes and favorable climate has made the city a destination for tech workers from across the country.

Key Facts About Austin Tech

  • Number of Tech Workers: 180,500; 13.7% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Dell, IBM, AMD, Apple, Alphabet
  • Key Industries: Artificial intelligence, hardware, cloud computing, software, healthtech
  • Funding Landscape: $4.5 billion in VC funding in 2024 (Pitchbook)
  • Notable Investors: Live Oak Ventures, Austin Ventures, Hinge Capital, Gigafund, KdT Ventures, Next Coast Ventures, Silverton Partners
  • Research Centers and Universities: University of Texas, Southwestern University, Texas State University, Center for Complex Quantum Systems, Oden Institute for Computational Engineering and Sciences, Texas Advanced Computing Center

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account