> Markdown version of [/jobs/ext/4585894-software-engineer-machine-learning-wfh](https://www.wearedevelopers.com/jobs/ext/4585894-software-engineer-machine-learning-wfh). Every page supports `.md` or `Accept: text/markdown`. Links point to the HTML versions so they work for humans too. Agent guide: [/agents.md](https://www.wearedevelopers.com/agents.md).

---

# Software Engineer, Machine Learning (WFH)

- **Company:** TRENT HOEK OUTDOORS LLC
- **Location:** San Francisco, CA, United States (Remote available)
- **Experience:** Experienced
- **Salary:** $153,000.0
- **Contract:** Permanent contract
- **Skills:** Artificial Intelligence, C++ (Programming Language), Software Code Optimization, Continuous Integration, Python (Programming Language), Machine Learning, Natural Language Processing, Tensorflow, Software Engineering, AI Infrastructure, Pytorch, Retrieval-Augmented Generation, Spark Mllib, Scikit Learn, Xgboost, Figma, Machine Learning Operations, Data Pipelines, Software Library, Programming Languages
- **Published:** November 5, 2026
- **Apply:** https://arc.dev/remote-jobs/j/redirect/pwodmx10n0

## About the Role

* At least five years of industry experience in software engineering, including at least three years focused on applied machine learning or AI.
 * Strong experience with end-to-end machine learning model development, including training, evaluation, deployment, and monitoring.
 * Proficiency in Python.
 * Familiarity with machine learning libraries such as PyTorch, TensorFlow, Scikit-learn, Spark MLlib, or XGBoost.
 * Experience designing and building scalable data pipelines, annotation workflows, and evaluation systems for AI model quality.
 * Experience mentoring or leading others and contributing to a culture of technical excellence and innovation., * Familiarity with search relevance, ranking, natural language processing (NLP), or RAG systems.
 * Experience with AI infrastructure and MLOps, including observability, continuous integration and continuous delivery (CI/CD), and automation for machine learning workflows.
 * Experience developing machine learning applications for creative or design-focused products.
 * Knowledge of additional programming languages, such as C++ or Go.
 * A product-oriented mindset and the ability to connect technical work with user outcomes and business impact.
 * Strong collaboration and communication skills, particularly when working across engineering, product, and research teams.

## Description

Figma is seeking a versatile and experienced Machine Learning / AI Engineer to join its growing AI team. This role sits at the intersection of applied machine learning, infrastructure, and product innovation, with opportunities to develop intelligent search systems, scalable data pipelines, and AI-powered creative tools.

You will collaborate with engineers, AI researchers, designers, and product managers across multiple teams to deliver high-quality machine learning features and infrastructure. Your work will contribute to foundational AI systems, improve user productivity, and shape new product experiences across the Figma platform.

This is a full-time position that can be based at one of Figma's US hubs or performed remotely within the United States., * Design, build, and productionize machine learning models for search, discovery, ranking, retrieval-augmented generation (RAG), and generative AI features.

 * Build and maintain scalable data pipelines to collect high-quality training and evaluation datasets.
 * Develop annotation systems and human-in-the-loop workflows to improve data quality and model performance.
 * Collaborate with AI researchers to refine datasets, evaluation metrics, and model architectures to improve model quality and relevance.
 * Partner with product engineers to define, develop, and deliver impactful AI features across Figma's platform.
 * Work with infrastructure engineers to develop and optimize systems for model training, inference, monitoring, and deployment.
 * Explore emerging technical approaches and contribute to Figma's long-term AI vision.
 * Contribute to technical excellence, innovation, and knowledge sharing across the engineering organization.