Lily AI · 12 hours ago
Staff Machine Learning Engineer
United States
Full-time
Remote
Senior Level, Lead/Staff
$140K/yr - $220K/yr
8+ years expLily AI is a female-founded retail AI company empowering retailers and brands by bridging the gap between merchant-speak and customer-speak. As a Staff Machine Learning Engineer, you will design and develop scalable platforms and services focused on driving business impact while collaborating with machine learning scientists, engineers, and product managers to define the ML roadmap and contribute to the AI strategy of the company.
Artificial Intelligence (AI)E-CommerceFashion
Responsibilities
Define, design, and maintain scalable Machine Learning data pipelines, training infrastructure, and inference systems
Optimize, benchmark, and productionize deep learning models to extract high-value product attributes, that drive revenue up in onsite search and google ads offerings
Drive cost efficiency and throughput improvements, owning relevant KPIs
Shape and evolve the technical stack to meet emerging business and technical needs
Transition research prototypes into robust, production-ready systems
Deploy, monitor, and continuously improve models in production environments
Optimize model performance, focusing on memory usage and latency
Automate workflows by building efficient pipelines and orchestration frameworks
Develop tools and shared libraries to boost team productivity and accelerate development
Qualification
Required
8+ years in building large-scale machine learning solutions and ML Ops practices
Working with LLM APIs and serving LLMs in-house at scale
Kubernetes, RDBMS, and API-driven development
Model serving in low-latency, high-throughput use cases
Observability, data pipeline design, service scaling, and cost optimization
Strong emphasis on code hygiene, including review, documentation, testing, and CI/CD practices
Proficiency in Python and PyTorch
Extensive experience with the scientific Python ecosystem
Proficiency in cloud-native application development
Action-oriented with the ability to articulate complex concepts into thoughtful, actionable iterations
Preferred
Proficient in writing high-performance production code in Python, with experience using frameworks like PyTorch
Excellent communication and interpersonal skills
Experience with Azure
Expertise in deep learning-based Computer Vision and NLP models
Proficiency with tools for managing the ML lifecycle, such as MLFlow and Kubeflow
Proficiency with real-time serving and optimization tools for deep learning, including TFX, PyTorch JIT, TorchScript, and Seldon
Master's degree in Computer Science or a related field
Company
Lily AI
Lily AI is the platform that bridges the gap between brand-speak and customer-speak to drive growth for commerce companies.
51-200 employees
H1B Sponsorship
Lily AI has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (4)
2024 (6)
2023 (7)
2022 (9)
2021 (7)
Funding
Current Stage
Growth StageTotal Funding
$71.9MKey Investors
Conductive VenturesCanaan PartnersUnshackled Ventures
2024-03-14Series B· $20M
2023-12-26Series Unknown· $12.4M
2022-06-09Series B· $25M
Leadership Team
Purva Gupta
Co-founder & CEO
Sowmiya Chocka Narayanan
Co Founder & CTO
Recent News
2025-08-14
2025-07-26
Company data provided by crunchbase