Machine Learning Engineer III - MLOps

Jumio Corporation; Kolkata District; On-site; INR 2,500,000 - 4,000,000; Full time

Job summary

Jumio, a leader in online identity verification, seeks a Software Development Engineer (SDE III) in the MLOps team in India. You will design scalable ML model serving, improve asset lifecycle management, and drive governance.

You will work with MLEs, QA, DevOps, and leverage MLFlow, PyTorch, and AWS serverless to optimize latency and cost while improving developer experience.

Qualifications

• 5+ years of software engineering experience in Python.
• Experience with ML model lifecycle management (MLFlow, Weights & Biases).
• Experience with data management ecosystem (quality, transformation, catalog).
• Experience with ML frameworks, particularly PyTorch.
• Experience optimizing ML models with hardware acceleration (AWS Neuron, ONNX, TensorRT).
• Proven experience building and operating AWS serverless architectures.
• Deep understanding of event-driven processing patterns, SQS/SNS and serverless caching solutions.
• Experience with containerization using Docker and orchestration tools.
• Strong knowledge of RESTful API design and implementation.
• Proficiency in writing good quality & secure code and be familiar with static code analysis tools.
• Excellent analytical, conceptual and communication skills in spoken and written English.
• Experience applying Computer Science fundamentals in algorithm design, problem solving, and complexity analysis.

Responsibilities

• Upgrade ML assets (models, data) management systems for better developer experience and robust governance capabilities
• Build and optimize model serving infrastructure with a focus on inference latency and cost optimization
• Architect efficient inference pipelines that balance latency, throughput, and cost across various acceleration options
• Implement cost-efficient, enterprise-scale solutions
• Collaborate in a cross-functional, distributed team for continuous system improvement
• Work with MLEs, QA Engineers, and DevOps Engineers
• Evaluate and implement new technologies and tools
• Contribute to architectural decisions for distributed ML systems

Skills

Python

PyTorch

RESTful API design

Algorithms & problem solving

English communication

Tools

MLFlow

Weights & Biases

Docker

AWS Neuron

ONNX

TensorRT

SQS/SNS

Job description

Role Purpose:

At Jumio, you will work for one of the market leaders in the global identity verification space that is helping to make the digital world a safer place for everyone. As a Software Development Engineer in the MLOpsTeam, you will develop the blueprint for highly scalable and performant ML model serving.

Role Value:

As a Software Engineer (SDE III), you will drive the continuous improvement of the infrastructure and applications to manage the lifecycle of ML assets (data, models) to better developer experience and strengthen governance capabilities. Secondly, you will design and implement robust ML infrastructure for model deployment, serving, and optimization. You will work on efficient CI/CD pipelines for ML models and leverage advanced compilers or hardware optimization to maximize inference performance while optimizing costs. We welcome you to challenge us to impact our software development processes and tools.

Example Responsibilities:

• Upgrade ML assets (models, data) management systems for better developer experience and robust governance capabilities
• Build and optimize model serving infrastructure with a focus on inference latency and cost optimization
• Architect efficient inference pipelines that balance latency, throughput, and cost across various acceleration options
• Implement cost-efficient, enterprise-scale solutions
• Collaborate in a cross-functional, distributed team for continuous system improvement
• Work with MLEs, QA Engineers, and DevOps Engineers
• Evaluate and implement new technologies and tools
• Contribute to architectural decisions for distributed ML systems

Experience and Qualifications:

• 5+ years of experience in software engineering with Python
• Experience with model lifecycle management (MLFlow, Weights & Biases or equivalent)
• Experience with data management ecosystem (quality, transformation, catalog)
• Experience with ML frameworks, particularly PyTorch
• Experience optimizing ML models with hardware acceleration (AWS Neuron , ONNX, TensorRT)
• Proven experience building and operating AWS serverless architectures
• Deep understanding of event-driven processing patterns, SQS/SNS and serverless caching solutions
• Experience with containerization using Docker and orchestration tools
• Strong knowledge of RESTful API design and implementation
• Proficiency in writing good quality & secure code and be familiar with static code analysis tools
• Excellent analytical, conceptual and communication skills in spoken and written English
• Experience applying Computer Science fundamentals in algorithm design, problem solving, and complexity analysis

Great to have Experience and Qualifications:

• Experience with any of the following: model compilation and quantization, performance profiling and benchmarking ML inference systems
• Experience working in regulated industries with strict compliance requirements for cloud-native solutions

Key Characteristics and Attitudes:

In a recent global survey these attributes were valued by Jumios in all locations and functions - we firmly believe in hiring for attitude as well as skill.

• Friendly and supportive
• Adaptable and flexible
• Articulate and persuasive
• High IQ and EQ
• Curious and coachable
• Commercially Aware
• Resilient and tenacious
• Big picture and the detail

Jumio Values:

IDEAL: Integrity, Diversity, Empowerment, Accountability, Leading Innovation

Equal Opportunities:

Jumio is a collaboration of people with different ideas, strengths, interests and cultures. We welcome applications and colleagues from all backgrounds and of all statuses.

About Jumio:

Jumio is a B2B technology company dedicated to eradicating online identity fraud, money laundering and other financial crimes to help make the internet safer. We leverage AI, biometrics, machine learning, liveness detection and automation to create solutions that are trusted by leading brands worldwide and respected by industry thought leaders.

Jumio is the leading provider of online identity verification, eKYC and AML solutions. With a global footprint, we’re expanding the team to meet strong client demand across a range of industries including Financial Services, Travel, Sharing Economy, Fintech, Gaming, and others.

Applicant Data Privacy:

We will only use your personal information in connection with Jumio’s application, recruitment, and hiring processes, as described in Jumio’s Applicant Privacy Notice. If you have any questions or comments, please send an email to [e-mail].