- Company Name
- Figma
- Job Title
- Software Engineer, Machine Learning
- Job Description
-
**Job Title**
Software Engineer, Machine Learning
**Role Summary**
Design, build, and production‑scale machine learning systems for Figma’s search, ranking, retrieval‑augmented generation, and generative AI features. Deliver end‑to‑end ML solutions, from data ingestion and annotation to model training, evaluation, deployment, and monitoring, while collaborating closely with product, research, engineering, and infrastructure teams.
**Expectations**
- Minimum 5 years of software engineering experience, with 3+ years focused on applied machine learning or AI.
- Proven track record of end‑to‑end ML model development: architecture design, training, evaluation, deployment, and monitoring.
- Proficiency in Python and mainstream ML libraries (PyTorch, TensorFlow, Scikit‑learn, Spark MLlib, XGBoost).
- Experience building and maintaining scalable data pipelines and annotation workflows.
- Demonstrated ability to mentor or lead technical teams and influence product outcomes.
**Key Responsibilities**
1. Design, implement, and productionise ML models for Search, Discovery, Ranking, Retrieval‑Augmented Generation, and generative AI features.
2. Build scalable data pipelines for training and evaluation, including annotation systems and human‑in‑the‑loop workflows.
3. Collaborate with AI researchers to iterate on datasets, evaluation metrics, and model architectures.
4. Partner with product engineers to define, prototype, and launch AI‑powered platform features.
5. Work with infrastructure engineers to optimise training, inference, monitoring, and deployment pipelines.
6. Contribute to the long‑term AI vision and explore innovative technical solutions.
**Required Skills**
- Strong programming in Python; familiarity with PyTorch, TensorFlow, Scikit‑learn, Spark MLlib, XGBoost.
- End‑to‑end ML model development and deployment expertise.
- Experience designing scalable data and annotation pipelines; knowledge of evaluation systems.
- MLOps skills: CI/CD, observability, automation of ML workflows.
- Ability to lead or mentor within a technical team.
- Excellent communication and cross‑functional collaboration.
**Required Education & Certifications**
Bachelor’s or Master’s degree in Computer Science, Software Engineering, Data Science, or an equivalent field. Professional certifications in ML or data engineering are a plus but not mandatory.