ML Engine Software Engineer

Who we are

Lightricks, an AI-first company, is revolutionizing how visual content is created. With a mission to bridge the gap between imagination and creation, Lightricks is dedicated to bringing cutting-edge technology to the creative and business spaces.

Our AI photo and video generation models, which power our apps and platforms including Facetune, Photoleap, Videoleap, and LTX Studio, allow creators and brands to leverage the latest research breakthroughs, offering endless control over their creative potential. Our influencer marketing platform, Popular Pays, provides creators the ability to monetize their work and offers brands opportunities to scale their content through tailored creator partnerships.

The ML Engine team is dedicated to driving innovation in machine learning (ML) by enhancing model efficiency and accelerating development cycles. This work is essential to keeping Lightricks at the forefront of the competitive generative AI landscape. The team focuses on developing and optimizing backend systems that support ML inference, directly impacting our products' speed and scalability.

What you will be doing

As ML Engine SW Engineer, you will play a crucial role in shaping the system design, backend architecture, and overall performance of Lightricks’ ML inference systems. We collaborate with cross-functional teams, including designers, developers, and researchers. The role offers the chance to make a significant impact on Lightricks’ core production systems. You’ll have the opportunity to develop and optimize critical backend features and work across diverse technical areas, contributing directly to the company's success. This hands-on role involves working extensively with Python, ComfyUI, system optimizations, and cloud infrastructure.

This position is perfect for a skilled backend engineer passionate about driving efficiency and innovation in machine learning. Join us to make an impactful difference in Lightricks’ AI journey!

Core Responsibilities

• Develop and maintain Lightricks' ML inference engine
• Drive improvements in system efficiency, monitoring, and performance
• Work directly with internal engine users to gather requirements and feedback
• Implement enhancements and optimizations within ComfyUI and other backend frameworks

Day-to-Day Responsibilities

• Design and optimize backend systems to support efficient ML model inference
• Work with cloud infrastructure, networking, and both CPU and GPU architecture
• Enhance the efficiency and observability of our inference engine, supporting various high-demand applications
• Collaborate closely with engine users, designers, developers, and researchers to refine and improve system capabilities
• Contribute to the ComfyUI codebase, ensuring a scalable, user-friendly interface for machine learning operations
• Navigating and contributing to a large, complex existing codebase
• Addressing complex issues with creativity and efficiency

Your skills and experience

• 5+ years of experience with Python in production environments
• Strong understanding of system architecture principles
• Proven familiarity with machine learning frameworks and methodologies
• Independent leadership with a proactive approach to problem-solving
• Strong analytical skills for diagnosing and addressing complex technical challenges
• A collaborative, team-player mindset that thrives in cross-functional environments

Preferred Requirements

• Knowledge of CPU/GPU architecture and experience with low-level optimizations
• Experience with image processing and computer vision
• Experience contributing to large, open-source projects