CrawlJobs Logo

Software Engineer, Machine Learning

figma.com Logo

Figma

Location Icon

Location:
United States, San Francisco ▼

Category Icon
Category:
IT - Software Development

Job Type Icon

Contract Type:
Not provided

Salary Icon

Salary:

149000.00 - 350000.00 USD / Year
Save Job
Save Icon
Apply Position

Job Description:

Figma is seeking a versatile and experienced Machine Learning / AI Engineer to join our growing AI team, working at the intersection of applied machine learning, infrastructure, and product innovation. Whether you’re building intelligent search systems, crafting scalable data pipelines, or enhancing AI-powered creativity tools, your work will drive user productivity, shape new product experiences, and advance the state of AI at Figma. You’ll collaborate closely with engineers, researchers, designers, and product managers across multiple teams to deliver high-quality ML-driven features and infrastructure. This is a high-impact, cross-functional role where you’ll shape both foundational systems and user-facing capabilities.

Job Responsibility:

  • Design, build, and productionize ML 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, including annotation systems and human-in-the-loop workflows
  • Collaborate with AI researchers to iterate on datasets, evaluation metrics, and model architectures to improve quality and relevance
  • Work with product engineers to define and deliver impactful AI features across Figma’s platform
  • Partner with infrastructure engineers to develop and optimize systems for training, inference, monitoring, and deployment
  • Explore new ideas at the edge of what’s technically possible and help shape the long-term AI vision at Figma

Requirements:

  • 5+ years of industry experience in software engineering
  • 3+ years focused on applied machine learning or AI
  • Strong experience with end-to-end ML model development, including training, evaluation, deployment, and monitoring
  • Proficiency in Python and familiarity with ML libraries like PyTorch, TensorFlow, Scikit-learn, Spark MLlib, or XGBoost
  • Experience designing and building scalable data and annotation pipelines, as well as evaluation systems for AI model quality
  • Experience mentoring or leading others and contributing to a culture of technical excellence and innovation

Nice to have:

  • Familiarity with search relevance, ranking, NLP, or RAG systems
  • Experience with AI infrastructure and MLOps, including observability, CI/CD, and automation for ML workflows
  • Experience working on creative or design-focused ML applications
  • Knowledge of additional languages such as C++ or Go
  • A product mindset with the ability to tie technical work to user outcomes and business impact
  • Strong collaboration and communication skills, especially when working across functions (engineering, product, research)
What we offer:
  • equity
  • health, dental & vision benefits
  • retirement with company contribution
  • parental leave & reproductive or family planning support
  • mental health & wellness benefits
  • generous PTO
  • company recharge days
  • a learning & development stipend
  • a work from home stipend
  • cell phone reimbursement
  • annual bonus plan for eligible non-sales roles

Additional Information:

Job Posted:
December 08, 2025

Employment Type:
Fulltime
Work Type:
On-site work
Icon
View All Jobs In This Company
Job Link Share:
Copy Link Icon
Facebook Icon Twitter Icon LinkedIn Icon

Looking for more opportunities? Search for other job offers that match your skills and interests.

Briefcase Icon

Similar Jobs for Software Engineer, Machine Learning

New

Staff Machine Learning Engineer

We are seeking a Staff Machine Learning Engineer to join our Foundation AI team....
Location
Location
United States , Boston
Salary
Salary:
170000.00 - 230000.00 USD / Year
whoop.com Logo
Whoop
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Advanced degree (Master’s or Ph.D.) in Computer Science, Machine Learning, Electrical Engineering, or a related field, or equivalent professional experience
  • 7+ years of experience in applied ML, AI research, or large-scale modeling, with a track record of delivering production systems
  • Expertise in modern deep learning (e.g., transformers, state space models) and multimodal model training
  • Proficiency in Python and deep learning frameworks (e.g., PyTorch, TensorFlow)
  • Experience building and scaling large datasets and training large models in distributed compute environments
  • Strong applied experience with representation learning, self-supervised methods, and fine-tuning for downstream applications
  • Familiarity with MLOps best practices including model versioning, evaluation, CI/CD for ML, and cloud-based compute
  • Excellent communication skills and ability to collaborate cross-functionally with engineers, researchers, and product teams
  • Passion for WHOOP’s mission to improve human performance and extend healthspan through science and technology
Job Responsibility
Job Responsibility
  • Design, train, and optimize large-scale multimodal foundation models that integrate wearable sensor data, text, biomarkers, and behavioral data
  • Conduct applied research in self-supervised learning, representation learning, and downstream task fine tuning to advance WHOOP’s core model capabilities
  • Develop scalable, distributed training pipelines for large models on high-performance compute environments
  • Collaborate with MLOps, data engineering, and software engineering teams to operationalize models for production deployment, ensuring robustness, reproducibility, and observability
  • Partner with product and research teams to translate foundation model capabilities into downstream features that deliver meaningful member value
  • Contribute to the technical roadmap and architectural direction for foundation model development at WHOOP
  • Serve as a technical mentor for other data scientists, sharing best practices in deep learning, large-scale training, and multimodal data integration
  • Ensure models adhere to WHOOP’s standards for ethical, transparent, and privacy-preserving AI
What we offer
What we offer
  • competitive base salaries
  • meaningful equity
  • benefits
  • generous equity package
  • Fulltime
Read More
Arrow Right
New

Senior Machine Learning Engineer (Health)

WHOOP is an advanced health and fitness wearable, on a mission to unlock human p...
Location
Location
United States , Boston
Salary
Salary:
150000.00 - 210000.00 USD / Year
whoop.com Logo
Whoop
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Bachelor’s Degree in Computer Science, Data Science, Applied Mathematics, or a related field. Master’s preferred
  • 5+ years of professional experience as a Machine Learning Engineer or Software Engineer with focus on ML systems
  • Proven expertise working with time series data (wearable, physiological, or high-frequency sensor data strongly preferred)
  • Experience designing and deploying ML inference systems at scale: both real-time streaming and large-scale batch pipelines
  • Strong coding skills in Python (scientific stack) and SQL, with a track record of writing clean, production-quality code
  • Strong communication skills to collaborate across engineering, research, and product teams
  • Proven experience deploying and maintaining ML systems on cloud platforms (AWS or GCP)
  • Working familiarity with MLOps best practices: model versioning, CI/CD for ML, observability, and monitoring for inference systems
  • Ability to reason about and design for performance trade-offs (latency vs. throughput vs. cost) when building ML inference systems
  • Strong understanding of backend service development (APIs and service reliability) as it applies to serving ML models at scale
Job Responsibility
Job Responsibility
  • Create, improve, and maintain production services that provide analysis for health features in collaboration with Data Scientists and MLOps Engineers
  • Collaborate with Data Engineers to improve ML data pipelines, tooling, and validation systems that support robust model performance
  • Work alongside data scientists to translate research prototypes into production ML systems optimized for scale, latency, and cost efficiency
  • Collaborate with researchers and product teams to align model development with health insights and member impact
  • Participate in on-call rotations for data science services, ensuring uptime and performance in production environments
What we offer
What we offer
  • equity
  • benefits
  • Fulltime
Read More
Arrow Right
New

Machine Learning Engineer II - Training

As a Machine Learning Engineer II on our Training team, you will develop algorit...
Location
Location
United States , Boston
Salary
Salary:
125000.00 - 170000.00 USD / Year
whoop.com Logo
Whoop
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Bachelor's Degree in Mathematics, Statistics, Computer Science, or a related field
  • 2+ years of ML engineering, applied research, or a similar role
  • 2+ years experience applying advanced mathematical and statistical techniques
  • Experience deploying and maintaining production ML systems on cloud platforms (e.g., Kubernetes, AWS, GCP)
  • Familiarity with MLOps best practices and the ability to collaborate effectively with infrastructure teams on Docker, CI/CD workflows, model versioning, and observability tools
  • Proficiency in scientific Python and SQL
  • Excellent verbal and written communication skills
Job Responsibility
Job Responsibility
  • Design, train, and optimize machine learning algorithms for movement, exercise and training applications across diverse backend platforms
  • Collaborate closely with data scientists, ML Ops and software engineering teams to ensure reliable deployment, observability, and robust integration with the WHOOP ecosystem
  • Contribute to technical roadmap development and architectural decision-making for projects that you are involved in
  • Work closely with a team of data scientists in developing algorithms that power member-facing features
  • Work with Data Engineers to improve data pipelining, tooling for machine learning, and systems for quality and validation
  • Periodically serve as the on-call data scientist to respond in real time to incidents affecting production services
What we offer
What we offer
  • equity
  • benefits
  • Fulltime
Read More
Arrow Right
New

Robotics Machine Learning Engineer

In this role you will close the loop between human instruction and robot automat...
Location
Location
United States , Watertown
Salary
Salary:
90000.00 - 190000.00 USD / Year
Tutor Intelligence
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • Knowledge of machine learning models, particularly visual models with an emphasis on recent advanced models such as SAM, DINO, etc.
  • Strong software skills
  • Full stack knowledge, ability to not only work on machine learning models in python but also to spin up inference infrastructure, web ui, and perform database operations
  • General robotics knowledge & experience
Job Responsibility
Job Responsibility
  • Train, fine tune, and deploy visual and language foundation models across our stack
  • Manage data in databases and optimize the flow of information from our robots to maximize machine learning potential
What we offer
What we offer
  • generous equity
  • fully covered health + dental
  • unlimited PTO
  • Fulltime
Read More
Arrow Right
New

Senior Machine Learning Engineer

Join Truveta’s Intelligence Platform and Applications team to engineer the next ...
Location
Location
United States , Seattle
Salary
Salary:
165000.00 - 190000.00 USD / Year
truveta.com Logo
Truveta
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • 5+ years of experience building and deploying scalable, production-ready ML systems in a collaborative engineering environment
  • Proficiency with agentic AI frameworks (e.g., LangGraph, AutoGen, CrewAI) and understanding of interoperability protocols such as MCP and A2A
  • Hands-on experience fine-tuning and optimizing large language models (LLMs) or multimodal models, using techniques such as LoRA, PEFT, and TRL
  • Experience working with vector databases and embeddings, integrating FAISS, Pinecone, Chroma, or Azure AI Search into retrieval-augmented generation (RAG) or semantic-search pipelines
  • Strong software engineering fundamentals, with proficiency in Python and experience designing scalable systems in modern cloud environments (Azure, AWS, or GCP)
  • B.S. or M.S. in Computer Science, Artificial Intelligence, or a related technical field
Job Responsibility
Job Responsibility
  • Design and reason with agentic AI frameworks
  • Bring a deep understanding of LLM fundamentals
  • Work fluently with embeddings and vector stores
  • Excel at model fine-tuning
  • Think and build like engineers
  • Collaborate across boundaries
  • Demonstrate senior-level ownership
  • Understand evaluation deeply
  • Adapt and learn continuously
  • Act with purpose
What we offer
What we offer
  • Interesting and meaningful work for every career stage
  • Great benefits package
  • Comprehensive benefits with strong medical, dental and vision insurance plans
  • 401K plan
  • Professional development & training opportunities for continuous learning
  • Work/life autonomy via flexible work hours and flexible paid time off
  • Generous parental leave
  • Regular team activities (virtual and in-person)
  • Fulltime
Read More
Arrow Right
New

Machine Learning Engineer - LLMs & Generative AI

Truveta is the world’s first health provider led data platform with a vision of ...
Location
Location
United States , Seattle
Salary
Salary:
155000.00 - 175000.00 USD / Year
truveta.com Logo
Truveta
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • 6+ years of experience in software engineering or machine learning (3+ years with a PhD)
  • Experience designing and training LLMs or large-scale generative models (e.g., GPT, PaLM, LLaMA, Claude, Gemma)
  • Deep expertise in NLP, sequence modeling, and transformer architectures
  • Proficient in Python and ML libraries such as PyTorch or TensorFlow
  • strong engineering skills in building scalable ML pipelines
  • Experience with RL-based fine-tuning (e.g., Reinforcement Learning from Human Feedback) and evaluation of generative systems
  • Proven ability to lead technical projects and collaborate across teams
  • Bachelor's degree in Computer Science, Engineering, or a related technical field
Job Responsibility
Job Responsibility
  • Lead the development, training, and deployment of large language and multimodal foundation models tailored to clinical and biomedical domains
  • apply and refine state-of-the-art techniques such as supervised fine-tuning (SFT), reinforcement learning-based methods (e.g., RLHF, RLVR), parameter-efficient fine-tuning (PEFT), prompt tuning, and retrieval-augmented generation (RAG)
  • Collaborate cross-functionally with researchers, clinicians, and engineers to design ML-driven solutions that improve healthcare delivery and outcomes
  • Build scalable infrastructure for distributed training of large models (TPU/GPU clusters, multi-node orchestration)
  • Design and evaluate models for robustness, bias mitigation, factual consistency, and explainability in healthcare contexts
  • Stay current with the latest research in generative AI and contribute back to the community through publications and open-source initiatives
What we offer
What we offer
  • Interesting and meaningful work for every career stage
  • Great benefits package
  • Comprehensive benefits with strong medical, dental and vision insurance plans
  • 401K plan
  • Professional development & training opportunities for continuous learning
  • Work/life autonomy via flexible work hours and flexible paid time off
  • Generous parental leave
  • Regular team activities (virtual and in-person)
  • Additional compensation such as incentive pay and stock options
  • Fulltime
Read More
Arrow Right
New

Staff Machine Learning Engineer

Machine Learning Engineers at Rocket Money further our mission by building produ...
Location
Location
United States , San Francisco; Washington, D.C.; New York City; Silver Spring; Miami; Denver
Salary
Salary:
210000.00 - 260000.00 USD / Year
Truebill
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • 8+ years of professional experience in machine learning engineering or data science roles
  • Proven track record of designing and implementing ML systems at consumer tech scale and speed
  • Extensive hands-on experience integrating ML and AI methods into production workflows, including creating evaluation tooling and effective user feedback mechanisms
  • Experience with prompt engineering and management, creating robust systems for testing and optimizing LLM-based applications
  • Expert-level proficiency in Python, SQL, and at least a handful of common ML frameworks
  • Understanding of ML methods at a fundamental level
  • Master at taking ambiguous problems, creating clarity, and breaking down work into manageable chunks for implementation
  • Owned the development, launch, and maintenance for several scaled ML/AI powered product experiences
  • Understand basic software engineering and computer science fundamentals and have applied them at consumer grade scale to build ML powered products in production environments
  • Technical leader who can identify both emergent technical opportunities and gaps relative to best practice
Job Responsibility
Job Responsibility
  • Lead the architecture and development of complex AI and ML powered features across Rocket Money's product suite
  • Design, implement, and maintain robust evaluation frameworks
  • Develop novel new product experiences
  • Own end to end development and implementation of ML and AI product features in collaboration with cross-functional product development teams
  • Provide technical mentorship
What we offer
What we offer
  • Health, Dental & Vision Plans
  • Competitive Pay
  • 401k Matching
  • Unlimited PTO
  • Lunch daily (in-office only)
  • Snacks & Coffee (in-office only)
  • Commuter benefits (in-office only)
  • Bonus
  • Fulltime
Read More
Arrow Right
New

Senior Machine Learning Engineer

Machine Learning is a cornerstone at Taskrabbit, and we're looking for a seasone...
Location
Location
United States , New York; San Francisco
Salary
Salary:
148000.00 - 200000.00 USD / Year
taskrabbit.com Logo
Taskrabbit
Expiration Date
Until further notice
Flip Icon
Requirements
Requirements
  • BS, MS, or PhD in Computer Science, Statistics, Operations Research, or a related quantitative field
  • 3+ years of industry experience building and deploying high-quality, production-grade machine learning models and systems
  • Strong theoretical knowledge and hands-on experience in machine learning, particularly in areas like search, ranking, recommender systems, or NLP
  • Solid software engineering skills with proficiency in one or more programming languages, including Python
  • Experience with popular ML libraries like Scikit-learn, lightgbm, xgboost, TensorFlow, PyTorch, etc.
  • Proficiency in SQL is also required for writing complex queries and transforming data
  • Experience building REST API-based services
  • Experience with modern data and ML technologies, such as Docker, Kubernetes, Kafka, Airflow, data warehouses (eg snowflake, redshift or BigQuery), and data lakes
  • Excellent communication skills, with the ability to present complex findings and recommendations clearly to both technical and non-technical audiences
  • A passion for quickly learning new technologies and a drive to solve challenging problems
Job Responsibility
Job Responsibility
  • Model Development & Research: Research, design, and implement machine learning models to solve key business problems in areas like search ranking, recommendations, and content discovery
  • End-to-End ML Lifecycle: Own the entire lifecycle of ML models, including feature engineering, training, evaluation, deployment, and monitoring
  • Infrastructure & Scalability: Build scalable and reliable ML infrastructure and data pipelines that support reproducible feature engineering and machine learning model deployment in real-time, near real-time, and batch processes
  • Performance & Quality: Build monitoring services to understand data quality and model performance of complex systems, and collaborate with engineering and science teams to optimize existing algorithms for training and evaluation
  • Software Engineering Excellence: Independently solve complex problems, write clean, efficient, and sustainable code, and actively participate in code reviews, documentation, and the full software engineering lifecycle
What we offer
What we offer
  • Taskrabbit is a Hybrid Company
  • The People
  • The Diverse Culture
  • Taskrabbit offers our employees with employer-paid health insurance and a 401k match with immediate vesting for our US based employees
  • We offer all of our global employees generous and flexible time off with 2 company-wide closure weeks, Taskrabbit product stipends, wellness + productivity + education stipends, IKEA discounts, reproductive health support, and more
  • Fulltime
Read More
Arrow Right
Welcome to CrawlJobs.com –
Your Global Job Discovery Platform
At CrawlJobs.com, we simplify finding your next career opportunity by bringing job listings directly to you from all corners of the web. Using cutting-edge AI and web-crawling technologies, we gather and curate job offers from various sources across the globe, ensuring you have access to the most up-to-date job listings in one place.
Jobs by Category
Art and Entertainment Jobs Banking Jobs Call Center Jobs Construction Jobs Consulting Jobs Customer Service Jobs Education Jobs Energy Jobs Finance Jobs Gastronomy Jobs Health and Beauty Jobs Hospitality and Tourism Jobs Human Resources Jobs Insurance Jobs IT - Administration Jobs IT - Software Development Jobs Legal Jobs Logistics Jobs Manufacturing Jobs Marketing, Digital Marketing, SEO, SEM Jobs Media Jobs Nursing Jobs Office Administration Jobs Pharmacy Jobs Physical Work Jobs Public Relations Jobs Public Sector Jobs Purchasing Jobs Quality Control Jobs Real Estate Jobs Research and Development Jobs Sales Jobs
Job Positions Jobs by Country
United States Jobs United Kingdom Jobs India Jobs Australia Jobs Canada Jobs Poland Jobs Ireland Jobs Spain Jobs
Company
Blog About Us Contact
Follow Us
Facebook
X (Twitter)
LinkedIn
Instagram
© 2025 CrawlJobs Ltd. All Rights Reserved
Terms & Conditions Privacy Policy