Top Machine Learning Jobs in Sacramento, CA

As a Senior ML Engineer, you'll own ML-driven features from modeling to deployment, drive innovation in generative AI, and mentor junior members.

Lead Android-first development of growth experiments and features, using AI-native workflows to design, implement, instrument, and analyze A/B tests. Ship performant, reliable Android experiences at scale, mentor teammates, participate in on-call, and codify AI-native engineering playbooks for broader adoption.

Design, build, and operate ML infrastructure powering data, compute, artifacts, and orchestration across cloud and on-prem. Own backend services, storage, observability, security, and developer tools; collaborate with cloud/compute providers and lead reliability and scaling efforts.

The AI Strategist leads AI initiative ideation and planning, collaborates across departments to define use cases, and ensures alignment with strategic and regulatory goals.

Build and maintain MLOps pipelines, deploy and monitor ML models as scalable services, transform and integrate healthcare data (FHIR/HL7), ensure reproducible feature management, and uphold HIPAA/HITRUST security for production ML in clinical systems.

Build and re-architect Shipt's LLM-native search stack including retrieval and ranking systems. Design multi-task and multi-objective models, integrate personalization signals, develop hybrid dense/sparse retrieval, and apply generative retrieval and query expansion. Analyze search data, report findings to leadership, and partner with senior scientists and ML engineers to improve relevance, revenue trade-offs, and operational metrics.

The Senior Machine Learning Engineer at Grailed will enhance personalization and recommendations, develop AI/ML solutions, and collaborate across departments for data-driven insights.

Design, develop, and deploy production forecasting models and agentic AI solutions on GCP. Perform EDA, feature engineering, experiment design, and model interpretability. Build and integrate AI agents (LangChain, Vertex AI Agents) into quotation and forecasting workflows, maintain MLOps pipelines (Vertex AI Pipelines, Cloud Composer), and present results to stakeholders.

Translate cutting-edge ML and physics research into production-grade scientific code, architect and scale large-scale simulation and training pipelines on GPU infrastructure, optimize hardware-level performance, and support models throughout the product lifecycle to accelerate in silico drug and materials discovery.

Develop and deploy advanced ML solutions for OCR and NLP, optimizing models for performance and integrating them into production workflows while collaborating with cross-functional teams.

Lead a team of data scientists developing ML-driven products, managing infrastructure, training models, and ensuring production quality while collaborating with senior leadership.

As a Machine Learning Engineer, you will design, build, and deploy ML models for underwriting at Stripe Capital, ensuring performance while collaborating with multiple teams.

The role involves designing, training, and evaluating ML models, optimizing performance, and collaborating with teams to implement AI technologies in products.

No responsibilities provided in the job description.

Build and maintain backend and full-stack marketplace features while supporting ML Ops infrastructure (model deployment, feature stores, real-time pipelines, monitoring). Collaborate with data scientists to bring models to production, write well-tested code, address performance and scalability issues, and contribute to technical planning and cross-functional communication.

Build and maintain full-stack marketplace features and support ML Ops infrastructure (model deployment, feature stores, real-time pipelines, monitoring) on Chalk and Vertex AI. Collaborate with data scientists to productionize models, write well-tested code, address performance/scalability issues, and contribute to technical planning.

Lead development of scalable AI/ML capabilities for Mail Intelligence: build LLM- and ML-based personalization, data pipelines, tools, and observability. Mentor engineers, collaborate cross-functionally, run A/B tests, implement feedback loops, and balance cost/quality/speed tradeoffs to deliver production-grade solutions.

Paid 3-month remote contract for students or recent grads to build and test LLM-based solutions using the Token Factory inference platform. Assist senior SAs with prompt engineering, model selection, benchmarking, and inference optimization. Run performance and quality experiments and contribute to internal tooling and automation. Mentorship provided; strong performers may be considered for full-time roles.

Build and maintain a capacity management portal and observability features for ML training and inference. Partner with researchers and PMs to turn requirements into intuitive UIs, integrate backend APIs across schedulers and training frameworks, develop internal tooling, and drive architectural improvements and best practices to improve efficiency and model quality visibility.

As a Data Scientist/Machine Learning Engineer, you'll finetune models, improve data quality, add new signals, and push solutions to production.

Lead and grow the AI/ML Engineering team, manage core AI/ML platform development, ensure production-grade service operational excellence, and develop product strategies.

Own end-to-end ML lifecycle for pricing and underwriting: improve model accuracy and coverage, reduce infrastructure cost, deploy and monitor production models on AWS, run experiments/backtests, and collaborate with domain experts to inform model improvements and product APIs.

Lead MLOps for production ML lifecycle: deploy and serve models, build scalable reproducible pipelines for large protein-design workloads, implement CI/CD and observability, optimize GPU/accelerated compute, and enable self-service tooling for researchers.

Design, build, and productionize ML and AI systems (LLMs, embeddings, recommendations, forecasting). Develop end-to-end ML workflows, scalable data pipelines, semantic models, and CI/CD for ML. Collaborate with product and engineering, mentor engineers, and influence platform architecture and standards.

Lead technical pre-sales and enterprise rollouts by owning infrastructure strategy, PoCs, migrations (on‑prem to cloud), DMS/CRM integrations, InfoSec discussions, and stakeholder engagement with C-level sponsors to enable scalable deployments and upsell opportunities.