Staff Machine Learning Engineer

Latitude; Atlanta (GA); Hybrid; USD 163,000 - 343,000; Full time

Benefits offered by this job

Employee ownership

Health insurance

401k with matching

Disability insurance

Paid time off

Growth opportunities

Job summary

Omnissa, the AI-driven digital work platform, is hiring for an ML Engineer on the AI Platform Team. Location: Mountain View, CA or Atlanta, GA, hybrid work. You will design, deploy ML models, build data pipelines, and improve model performance at scale.

Requirements include 5+ years in ML engineering or data science, strong Python (PyTorch, TensorFlow, Scikit-learn), cloud experience (AWS/Azure/GCP), Docker/Kubernetes, and hands-on with LLMs and vector databases. Competitive USD salary.

Qualifications

• five+ years of experience in machine learning engineering or data science roles.
• Proficiency in Python and ML frameworks (PyTorch, TensorFlow, Sci-kit Learn).
• Experience building/operating data workflows (batch/streaming) and cloud platforms (AWS/Azure/GCP).
• Knowledge of ML algorithms, statistics, model evaluation, and drift detection.

Responsibilities

• Design, develop, and deploy ML models for classification, prediction, anomaly detection, and automation.
• Build and maintain scalable data pipelines for training, evaluation, and inference.
• Optimize ML models for performance, scalability, reliability, and cost.
• Collaborate with cross-functional teams to integrate ML into core features.
• Experiment and evaluate models using quantitative metrics and A/B tests.

Skills

Python

PyTorch

TensorFlow

Scikit-learn

LLMs

Docker

Kubernetes

Vector databases

RAG

Agile teams

Education

Bachelor's Degree

Tools

Docker

Kubernetes

LangChain

LangGraph

Spark

Ray

REST/gRPC

Job description

Job Description

Omnissa is the first AI-driven digital work platform, built to support flexible, secure, work-from anywhere experiences. We integrate industry-leading solutions—including Unified Endpoint Management, Virtual Apps and Desktops, Digital Employee Experience, and Security & Compliance—into a seamless, autonomous workspace that adapts to how people work. Our platform boosts employee engagement while optimizing IT operations, security, and cost.

We are Omnissa!

Guided by our Core Values—Act in Alignment, Build Trust, Foster Inclusiveness, Drive Efficiency, and Maximize Customer Value—we’re growing rapidly and committed to delivering meaningful impact. If you're passionate about shaping the future of work, we’d love to hear from you.

At Omnissa, we are committed to maintaining a fair, consistent, and secure hiring process for all candidates. As part of this approach, we use standard interview and verification practices designed to ensure alignment and protect both candidates and the organization. These practices are applied thoughtfully and with respect for candidate privacy.

What is the opportunity?

Our platform manages millions of devices across multiple operating systems, requiring exceptional performance, scalability, availability, and resilience. You will join the AI Platform Team, the group responsible for building foundational AI capabilities across the Omnissa product ecosystem.

Responsibilities

• Design, develop, and deploy machine learning models for classification, prediction, anomaly detection, and intelligent automation.
• Build and maintain scalable data pipelines for model training, evaluation, and real time/batch inference.
• Optimize ML models and pipelines for performance, scalability, reliability, and cost efficiency.
• Collaborate with cross functional teams to integrate ML solutions into core platform features and services.
• Conduct model experimentation, evaluation, and iteration using quantitative metrics and A/B testing as needed.
• Implement model observability, monitoring, and drift detection to ensure production reliability.
• Stay current with advancements in machine learning, AI, and LLM technologies, and apply them to product use cases.

What will you bring to Omnissa?

• 5+ years of experience in machine learning engineering or data science roles.
• Strong proficiency in Python and ML frameworks (e.g., PyTorch, TensorFlow, Scikitlearn).
• Experience building and operating data processing workflows (batch or streaming) and working with cloud platforms (AWS, Azure, or GCP).
• Solid understanding of machine learning algorithms, statistics, and model evaluation techniques.
• Familiarity with containerization and orchestration technologies (Docker, Kubernetes).
• Handson experience with Large Language Models (LLMs), including finetuning, prompt engineering, and deployment.
• Knowledge of text embedding models, and vector databases for Retrieval Augmented Generation (RAG) systems
• Strong problem-solving skills and the ability to collaborate effectively in Agile teams.
• Highly motivated, adaptable, and eager to learn new technologies.

Preferred Skills

• Experience with distributed computing frameworks (e.g., Spark, Ray).
• Experience with orchestration frameworks (e.g., LangChain/LangGraph) to build AI agents and multi-agent systems.
• Experience building feature stores or working with vector databases.
• Knowledge of real-time inference architectures and model monitoring systems.
• Experience developing scalable ML services via REST/gRPC.

Location

Mountain View, CA or Atlanta, GA

Location Type

hybrid

Travel Expectations

None

Education

Bachelor's Degree preferred, or equivalent combination of education and relevant professional experience.

Compensation

The typical base salary for this role is between USD $162,512 – $342,750 per year and it may be eligible for participation in a corporate bonus program. Actual compensation offer may vary from posted hiring range based upon geographic location, work experience, education, skill level, or other relevant factors. In addition to competitive compensation, Omnissa offers a variety of benefits such as employee ownership, health insurance, 401k with matching contributions, disability insurance, paid-time off, growth opportunities, and more.

Omnissa is an Equal Employment Opportunity company and Prohibits Discrimination and Harassment of Any Kind

Omnissa is committed to the principle of equal employment opportunity and to providing a work environment free of discrimination and harassment. All employment decisions at Omnissa are based on business needs, job requirements and individual qualifications, without regard to race, color, religion, ancestry, ethnicity, national, social or ethnic origin, sex (including pregnancy), age, physical, mental or sensory disability, HIV status, sexual orientation, gender identity and/or expression, marital, civil union or domestic partnership status, past, present, or prospective service in the uniformed services, family medical history or genetic information, family or parental status, veteran status, or any other status protected by applicable laws or regulations in the locations where we operate. Omnissa will not tolerate discrimination or harassment based on any of these characteristics. Omnissa welcomes applicants of all ages. Omnissa will provide reasonable accommodations to applicants and employees who have protected disabilities consistent with applicable federal, state and local law.