DaVita logo
DaVita

DaVita is a leader in quality care and education for chronic kidney disease and end-stage renal disease. Since 1999, the company has worked toward a mission to

Machine Learning Engineer

Machine Learning EngineerMachine Learning EngineerOtherRemoteSeniorTeam 225Since 2019Company Site

Location

United States

Posted

126 days ago

Salary

$150K - $215K / year

Seniority

Senior

Job Description

Machine Learning Engineer

DaVita

• Design and build scalable ML services for enrichment workflows, including model training pipelines and high-performance inference APIs • Deploy and optimize models using modern inference libraries and frameworks (ONNX, vLLM, TensorRT, etc.) to achieve low-latency, high-throughput performance • Collaborate with software engineers and product teams to define data requirements, feature engineering strategies, and model evaluation metrics • Build robust monitoring, observability, and evaluation systems to ensure model quality and service reliability in production • Stay current with emerging ML techniques, tools, and best practices, particularly in areas like model optimization, efficient inference, and large-scale data processing

Job Requirements

  • 5+ years of experience building and deploying machine learning systems in production environments
  • Strong proficiency with model deployment technologies (Kubernetes, Ray, etc.) and inference libraries (ONNX, vLLM, TensorRT, or similar). Proficiency with model training frameworks (PyTorch, TensorFlow, Jax)
  • You've successfully designed and scaled ML services that process large volumes of data and serve predictions with strict latency and throughput requirements
  • Experience with the full ML lifecycle, including data preprocessing, feature engineering, model training, evaluation, deployment, and monitoring
  • Solid software engineering skills, including experience with distributed systems, APIs, and cloud infrastructure
  • You have a passion for building reliable, performant ML systems and understand how they create value for end users
  • U.S. Person status is required as this position will require the ability to access U.S only data systems.

Benefits

  • Health, dental, and vision insurance
  • Remote friendly with WeWork access
  • Unlimited PTO, shared downtime during the federal holiday calendar, and company-wide off time at the end of each year
  • 401(k) match
  • Lifestyle & wellbeing stipends
  • Salary top-up during military reserve duty
  • Fully paid parental leave
  • Child and pet care reimbursement during travel

Related Job Pages

More Machine Learning Engineer Jobs

Correlation One logo

Content Developer – Machine Learning Annotation

Correlation One

Correlation One is a technology company that is on a mission “to create equal access to data-driven jobs of tomorrow.” As an employer, the company is known for its empowering,

• Create high-quality instructional materials that teach modern data annotation and labeling workflows used to train and evaluate machine learning systems. • Develop content across modalities—including text, image, video, audio/speech, and cross-modality labeling. • Collaborate with internal program team to produce session plans, hands-on practice, examples, and assessments aligned to defined learning outcomes. • Content should model best-in-class annotation behavior: rule-based decisions, clear rationales, consistent application of guidelines, and appropriate handling of ambiguity. • Produce complete content packages per session/module, such as facilitator guide, learner materials, demos or walkthroughs, practice activities, strong vs. weak examples, rubrics/evaluation criteria, and short knowledge checks/assessments. • Create hands-on exercises using a common annotation platform, including task setup guidance, labeling instructions, example labels, edge cases, and review workflows. • Review and iterate on content based on internal feedback and peer review. • Participate in weekly and ad hoc meetings with the program team to align on objectives, standards, and delivery constraints. • Support content deployment into our learning platform (training provided).

United States
Job Closed
PandaDoc logo

Machine Learning Engineer – Document Intelligence, Applied GenAI

PandaDoc

PandaDoc is a computer software company that is working to empower clients “to streamline their process” to negotiate, generate, and sign a variety of documents and provide the

• Build and maintain evaluation frameworks for document models, LLMs, OCR, and structured extraction. • Define metrics, benchmarks, and validation strategies for real-world document workloads. • Design and curate high-quality datasets for supervised training, fine-tuning, and validation. • Create scalable preprocessing pipelines for PDFs, scans, images, forms, and semi-structured documents. • Train and fine-tune transformer-based OCR, VLMs, layout models, and open-source LLMs for document understanding tasks. • Optimize models for reliability, accuracy, and cost efficiency in production environments. • Deploy ML models with modern inference runtimes (vLLM, TGI, TensorRT, ONNX Runtime). • Build guardrails, monitoring, and fallback mechanisms to ensure safe and predictable model behavior. • Develop retrieval and chunking strategies tailored to document structures (tables, forms, multi-page PDFs). • Optimize end-to-end RAG pipelines for semantic search, Q&A, and workflow automation. • Partner with PMs, backend engineers, and product designers to define AI opportunities and translate requirements into technical solutions.

Spain
Job Closed
PandaDoc logo

Machine Learning Engineer – Document Intelligence, Applied GenAI

PandaDoc

PandaDoc is a computer software company that is working to empower clients “to streamline their process” to negotiate, generate, and sign a variety of documents and provide the

• Build and maintain evaluation frameworks for document models, LLMs, OCR, and structured extraction. • Define metrics, benchmarks, and validation strategies for real-world document workloads. • Design and curate high-quality datasets for supervised training, fine-tuning, and validation. • Create scalable preprocessing pipelines for PDFs, scans, images, forms, and semi-structured documents. • Train and fine-tune transformer-based OCR, VLMs, layout models, and open-source LLMs for document understanding tasks. • Optimize models for reliability, accuracy, and cost efficiency in production environments. • Deploy ML models with modern inference runtimes (vLLM, TGI, TensorRT, ONNX Runtime). • Build guardrails, monitoring, and fallback mechanisms to ensure safe and predictable model behavior. • Develop retrieval and chunking strategies tailored to document structures (tables, forms, multi-page PDFs). • Optimize end-to-end RAG pipelines for semantic search, Q&A, and workflow automation. • Partner with PMs, backend engineers, and product designers to define AI opportunities and translate requirements into technical solutions.

Germany
Job Closed

Staff Machine Learning Engineer: Search

PrizePicks

PrizePicks is the fastest-growing sports company in North America according to the 2023 Inc. 5000 rankings, two years running, and the largest independent skill-based fantasy sports operator in the country.

OtherRemoteTeam 201-500H1B No Sponsor

This description is a summary of our understanding of the job description. Click on 'Apply' button to find out more. Role Description As a Staff Data Science Engineer, you will be developing, maintaining, testing, and leading projects regarding global search. You will architect a native Search & Discovery Module powered by our Dataverse Stack. Your goal is to deliver a search experience that is fast (<200ms), smart, and personalized. You will transform Search from a utility into a primary driver of gameplay. - Architect Global Search: - Build the foundational architecture for a unified Global Search entry point that creates a "Single Source of Truth" across Players, Teams, and Game Modes. - Optimize for Speed: - Engineer the Prizepicks search stack to achieve <200ms latency targets, evaluating and deciding between Server-side indexing. - Advanced Retrieval Logic: - Implement fuzzy matching, nickname support, and natural language processing (NLP) to handle user queries like "Bron", “lal/LAL/Lakers” or "Rushing Yards" intuitively. - Dataverse Integration: - Partner with Data Engineering to ensure the Search Index is fed by real-time streams of projections, live game states, and social data, ensuring results are never stale. Qualifications - 5+ years of experience in Machine Learning Engineering, with deep expertise in Information Retrieval (IR) and Search technologies. - 3+ years of technical leadership, guiding teams through complex architectural migrations and greenfield builds. - Familiarity with indexing strategies using technologies like Elasticsearch, OpenSearch, Solr, or Vector Databases (Pinecone, Milvus). - A track record of optimizing API response times and database queries for high-throughput, low-latency applications. - Experience with GCP (Kubernetes, Cloud Functions) and Infrastructure as Code (Terraform). Requirements - Experience building "Instant Search" or "Type-ahead" features for high-traffic consumer mobile apps. - Knowledge of mobile-side database technologies (e.g., Realm, SQLite, WatermelonDB) for offline-first or hybrid search architectures. - Experience integrating GenAI or LLMs to power conversational search interfaces. Benefits - Company-subsidized medical, dental, & vision plans - 401(k) plan with company match - Annual bonus - Flexible PTO to encourage a healthy work/life balance (2 weeks STRONGLY encouraged!) - Generous paid leave programs, including 16-week paid parental leave and disability benefits - Workplace flexibility and modern work schedules focused on getting the job done, not hours clocked - Company-wide in-person events and team outings - Lifestyle enhancement program - Company equipment provided (Windows & Mac options) - Annual performance reviews with opportunities for growth and career development

United States
Job Closed