Senior Machine Learning Engineer

Company Description

CodeFlamme is committed to turning innovative ideas into reality through cutting-edge digital transformation solutions. The company emphasizes creativity, technology, and business growth, delivering exceptional tech excellence for its clients. At CodeFlamme, we empower businesses by providing reliable and impactful digital solutions. Join us to make a difference in the world of #DigitalTransformation and #Innovation.

Role Description

This is a full-time, on-site role for a Senior Machine Learning Engineer based in Lahore. In this position, you will design and implement advanced machine learning models and algorithms, analyze complex datasets, and develop innovative solutions for diverse projects. Collaborating with cross-functional teams, you will employ state-of-the-art techniques such as diffusion models, neural networks, and pattern recognition to solve real-world problems while optimizing performance, scalability, and efficiency.

Qualifications

• Strong expertise in Machine Learning fundamentals, including Pattern Recognition and Neural Networks
• Proficiency in Computer Science concepts, including Algorithms and Data Structures
• Solid understanding of Statistics and Mathematical foundations critical for data modeling
• Excellent analytical and problem-solving skills
• Experience working in a collaborative environment with cross-functional teams
• Bachelor’s or Master’s degree in Computer Science, Data Science, or a related field
• Previous experience with large-scale machine learning projects is a plus

Core Technical Skills

• Diffusion Models (Stable Diffusion, SDXL, DALL·E architecture)
• Null-text inversion techniques
• Latent-space manipulation & editing
• Attention mechanisms & cross-attention control
• PyTorch (required), TensorFlow (nice-to-have)
• SSIM/LPIPS metrics & quality optimization
• GPU optimization (A100/H100 inference tuning)

Experience Requirements

• 3+ years of professional experience in Machine Learning and Computer Vision
• Proven experience in shipping production diffusion model features

Job Type: Full-time

Pay: From Rs80,000.00 per month

Work Location: In person