SDE III - MLOps

Jumio seeks a Software Development Engineer in MLOps to build scalable ML model serving infrastructure, manage the lifecycle of ML assets, and optimize inference performance. You will work with cross-functional teams on CI/CD pipelines, hardware acceleration, and serverless architectures.

• 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

• 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 familiarity with static code analysis tools
• Excellent analytical, conceptual, and communication skills in English
• Experience applying Computer Science fundamentals in algorithm design, problem solving, and complexity analysis

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