The Home Depot · 3 hours ago
Staff Machine Learning Engineer
Atlanta, GA
Full-time
Onsite
Senior Level, Lead/Staff
$120K/yr - $190K/yr
3+ years expThe Home Depot is seeking a Staff Machine Learning Engineer who will lead a team of engineers in building and designing customer-centric products. This role involves providing technical leadership on machine learning systems, collaborating with cross-functional teams, and ensuring the development of reliable and scalable ML solutions.
RetailE-CommerceHome ServicesHome and GardenHome DecorHome Improvement
Responsibilities
Collaborates and pairs with other product team members (UX, engineering, and product management) to create secure, reliable, scalable machine learning solutions
Works with Product Team to ensure user stories that are developer-ready, easy to understand, and testable
Configures commercial off the shelf solutions to align with evolving business needs
Creates meaningful dashboards, logging, alerting, and responses to ensure that issues are captured and addressed proactively
Participates in learning activities around modern software design, machine learning, and development core practices (communities of practice)
Proactively views articles, tutorials, and videos to learn about new technologies and best practices being used within other technology organizations
Attends conferences and learns how to apply new innovations and technologies where appropriate
Researches and analyzes business trends and behavioral data to identify opportunities for improvement and new initiatives
Leads the evaluation development and recommendation of specific technology products and platforms to provide cost-effective solutions that meet business and technology requirements
Researches and designs best fit infrastructure, network, database, security, and machine learning architectures for products
Proactively creates and maintains tools for monitoring and support
Participates in project planning and management across multiple efforts
Develops formal training courses
Fields questions from other product teams or support teams
Monitors tools and participates in conversations to encourage collaboration across product teams
Provides application support for software running in production
Proactively monitors production Service Level Objectives for products
Proactively reviews the Performance and Capacity of all aspects of production: code, infrastructure, data, message processing, and prediction quality
Qualification
Required
Must be eighteen years of age or older
Must be legally permitted to work in the United States
3 years of relevant work experience
Strong experience designing, training, evaluating, and deploying machine learning models in production environments, including batch and real-time inference systems
Experience with ML lifecycle management, including feature engineering, model versioning, experimentation, validation, and monitoring for data drift and model performance degradation
Proficiency with common ML frameworks and tools (TensorFlow, PyTorch, scikit-learn) and experience integrating them into scalable software systems
Experience building and operating ML pipelines using cloud-native services, data platforms, and CI/CD practices for reproducible and reliable model deployment
Strong understanding of applied statistics, model evaluation metrics, and tradeoffs between model accuracy, interpretability, latency, and operational cost
Experience to algorithms such as clustering, forecasting, anomaly detection, and neural networks
Experience to basic statistics and regression algorithms
Experience in advanced machine learning techniques such as NLP, convolutional neural networks, autoencoders, and embeddings generation and utilization
Experience in training machine learning models with extremely large datasets
Experience with Data Analysis and Machine Learning Tools and Libraries like Jupyter Notebooks, Pandas, SciPy, Scikit-learn, Gensim, tensorflow, pytorch, etc
Experience with GPU acceleration (i.e. CUDA and cuDNN)
Experience in Google Cloud Platform and AI/ML related components such as Vertex AI, BigQueryML, and AutoML
Experience in effective data engineering practices and big data platforms such as BigQuery, Data Store, etc
Experience in a modern scripting language (preferably Python)
Experience in modern web application framework such as Node.js
Experience in a front-end technology and framework such as HTML, CCS, JavaScript, ReactJS, D3
Experience in writing SQL queries against a relational database
Experience in version control systems (preferable Git)
Experience in a Linux or Unix based environment
Experience in a CI/CD toolchain
Experience in REST and effective web service design
Experience in production systems design including High Availability, Disaster Recovery, Performance, Efficiency, and Security
Experience in NoSQL databases
Experience in cloud computing platform and associated automation patterns and machine learning services they provide
Experience in defensive coding practices and patterns for high Availability
Experience in A/B testing and effective REST design for scalable web services architecture
Familiarity with advanced machine learning architectures GANs, GRU, LSTMs, RNNs, CNNs, style transfer
Preferred
3 - 6 years of relevant work experience
No additional education
No additional years of experience
None
None
Company
The Home Depot
The Home Depot is a retailer of hardware, consumer goods, and DIY supplies both online and offline while providing home delivery services.
10001+ employees
H1B Sponsorship
The Home Depot 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 (32)
2024 (67)
2023 (32)
2022 (66)
2021 (62)
2020 (33)
Funding
Current Stage
Late StageTotal Funding
unknown2007-07-31Debt Financing
Leadership Team
Ted Decker
Chair, President and CEO
Chip Devine
Senior Vice President
Recent News
2026-02-06
Mexico Business
2026-02-02
Company data provided by crunchbase