1

Machine Learning Software Engineer Jobs in San Jose, CA

Who You Are We're looking for innovative and passionate Machine Learning Engineers to join our team ... Good understanding of software development principles, data structures, and algorithms. * Excellent ...

Minimum Qualifications Software engineering skills and proficiency in Python Experience with ... machine learning, computer science, computer engineering or related fields.

Software engineering skills and proficiency in Python. Experience with PyTorch. BA/BS degree in computer vision, computer graphics, machine learning or related field. Preferred Qualifications MS or ...

Software Engineer, Machine Learning Responsibilities: * Collaborate with cross-functional teams (product, design, operations, infrastructure) to build innovative application experiences * Implement ...

next page

Showing results 1-20

Machine Learning Software Engineer information

See San Jose, CA salary details

$74.4K

$172.9K

$240.8K

How much do machine learning software engineer jobs pay per year?

As of Jul 19, 2026, the average yearly pay for machine learning software engineer in San Jose, CA is $172,896.00, according to ZipRecruiter salary data. Most workers in this role earn between $140,600.00 and $202,800.00 per year, depending on experience, location, and employer.

What does a Machine Learning Software Engineer do?

A Machine Learning Software Engineer designs, develops, and deploys machine learning models within software applications. They work on data preprocessing, model training, optimization, and integration into production systems. Their role requires expertise in programming (Python, Java, or C++), machine learning frameworks (TensorFlow, PyTorch, or Scikit-learn), and cloud platforms. They collaborate with data scientists and software engineers to build scalable ML solutions.

What are the key skills and qualifications needed to thrive in the Machine Learning Software Engineer position, and why are they important?

To thrive as a Machine Learning Software Engineer, you need a solid understanding of programming (especially Python), algorithms, data structures, and mathematics, ideally backed by a degree in computer science, engineering, or a related field. Experience with frameworks such as TensorFlow or PyTorch, familiarity with cloud platforms (AWS, Azure, or GCP), and relevant certifications in data science or machine learning are highly valuable. Strong problem-solving skills, effective communication, and the ability to work collaboratively with cross-functional teams set outstanding candidates apart. These competencies are crucial for building deployable, scalable, and maintainable machine learning solutions that address real business challenges.

What are the day-to-day responsibilities of a Machine Learning Software Engineer?

As a Machine Learning Software Engineer, your daily tasks typically include developing and optimizing machine learning models, collaborating with data scientists and product teams to define requirements, and integrating models into production systems. You’ll work extensively with large datasets to preprocess, analyze, and validate data, as well as monitor model performance and iterate on solutions when needed. It's common to participate in code reviews, contribute to architectural decisions, and maintain documentation for reproducibility and knowledge sharing. This role offers a dynamic and intellectually stimulating environment, making it ideal for those who enjoy solving complex technical problems and working at the intersection of engineering and data science.

What cities near San Jose, CA are hiring for Machine Learning Software Engineer jobs? Cities near San Jose, CA with the most Machine Learning Software Engineer job openings:
Software Engineer, Machine Learning

Software Engineer, Machine Learning

Figma

San Francisco, CA • On-site, Remote

Other

Re-posted 22 days ago


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.

This is a full time role that can be held from one of our US hubs or remotely in the United States.

What you'll do at Figma:
  • 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.
We'd love to hear from you If you have:
  • 5+ years of industry experience in software engineering, with 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.
While not required, It's an added plus if you also 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 is a plus, but not required.
  • 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).
At Figma, one of our values is Grow as you go. We believe in hiring smart, curious people who are excited to learn and develop their skills. If you're excited about this role but your past experience doesn't align perfectly with the points outlined in the job description, we encourage you to apply anyways. You may be just the right candidate for this or other roles.