Cognizant

Staff Machine Learning Infrastructure Engineer

Cognizant
Apply
12 hours ago

Base Salary

$130k - $145k/yr

Responsibilities

  • Architect and evolve end-to-end machine learning platforms supporting data processing, feature management, training, and model serving.
  • Design and implement scalable infrastructure for the development, deployment, and operation of ML workloads.
  • Drive technical strategy, standards, automation, best practices, and platform reliability for ML infrastructure.
  • Lead real-time and batch inference solutions supporting high-volume production workloads.
  • Mentor engineers, lead technical reviews, and collaborate with data science and engineering teams to expand ML adoption.

Requirements

  • 10+ years of software engineering experience, including 5+ years focused on machine learning infrastructure, MLOps, or platform engineering.
  • Deep expertise designing and operating distributed systems and large-scale data processing platforms.
  • Proven experience building and supporting production ML platforms for mission-critical business applications.
  • Hands-on experience with Apache Spark, Apache Beam, feature stores, and large-scale data pipelines.
  • Strong knowledge of ML serving and inference platforms such as TorchServe, TensorFlow Serving, and Triton.
  • Experience with containerization, orchestration platforms, cloud-native architectures, and infrastructure automation.
  • Demonstrated success leading complex technical initiatives and influencing architecture decisions across engineering organizations.
  • Preferred qualifications include GPU-based ML infrastructure, high-performance computing, ML observability, monitoring, reliability engineering, open-source contributions, and ML security, governance, or compliance experience.
  • A master's degree in computer science, engineering, or a related technical field is preferred.

Benefits

  • Onsite work five days per week in a client or Cognizant office in Sunnyvale, California.
  • Medical, dental, vision, and life insurance.
  • Paid holidays and paid time off.
  • 401(k) plan and contributions.
  • Long-term and short-term disability coverage.
  • Paid parental leave.
  • Employee Stock Purchase Plan.
  • Eligible for a discretionary annual incentive program based on performance.

Tech Stack

Apache BeamApache Spark
Cognizant

About Cognizant

10,000+ employees

Cognizant is a public IT services and consulting firm that designs, builds, and runs enterprise technology, including digital engineering, cloud modernization, data/AI, and managed services. It sells consulting, systems integration, and outsourcing on multi-year engagements to large enterprises in healthcare, banking, retail, communications, and manufacturing. Founded in 1994 and headquartered in Teaneck, New Jersey, Cognizant is NASDAQ-listed (CTSH) and a Fortune 500 company.

Contact me