AI/ML Engineer (US)

San Jose, California, United States | Engineering | Full-time

Apply by: No close date

Who are we?

Whatfix is a data-driven digital adoption platform (DAP) that enables organizations and users to maximize the benefits of software. Whatfix acts as an interactive overlay on top of any application to guide users with real-time guidance, self-help support, and user feedback. With product analytics and AI, Whatfix enables scalable success with technology, maximizing productivity, and leveraging data-driven insights for better decision-making. The company has seven offices globally in the US, India, UK, Germany, Singapore, and Australia, and works with Fortune 500 companies around the world. Whatfix has raised $140 million to date and is backed by marquee investors including Softbank, Sequoia, Dragoneer, and Cisco Investments. “Hustle Mode ON” is the motto we live by.

About the Role:
Whatfix is working on building bots/assistant that can take instructions by users in Natural language (e.g. "Apply for Casual leave from 20th Aug to 25th Aug 2023”) and the bot will control the browser and execute the task on any application on the web. We are looking for rockstar AI/ML Engineers who have worked on UIs and AI/ML.

This role is required to work onsite at our San Jose office. The daily working hours for this role begin at 6:30am PST and there is flexibility to work at home for a few hours & come into the office a bit later (e.g. after dropping off kids at school). Relocation assistance is offered.

Key Responsibilities:  

  • Develop advanced computer vision and state-of-the-art deep learning models for building an understanding of software applications and using the understanding to act as an agent
  • Research & Implement ML and Deep learning algorithms for production
  • Build infrastructure needed for AI/ML systems, such as model inference, automated (re-)training, monitoring, explainability, etc.
  • Participate in Code Review and process improvement
  • Stay abreast of new AI research from leading labs by reading papers and experimenting with code
  • Help AI product managers and business stakeholders understand the potential and limitations of AI for your Products
  • Run necessary ML tests and benchmarks for model validation
  • Fine-tune, retrain, and scale existing model deployments
  • Articulate complex problems by communicating viable architecture and design approaches in a clear and concise manner
  • Implement computationally complex model training and software applications understanding Neural networks and integrate them into the Whatfix SaaS platform
  • Design and develop highly scalable, reliable, and fault-tolerant systems for one of the world’s fastest-growing startups
  • Envision project goals, and product architectural solutions with a high focus on detail to ensure successful implementation
  • Occasionally participate in customer demos & provide inputs wherever required
  • Conduct research to provide technical solutions at scale for real-world challenges in various scenarios

Our ideal AI/ML Engineer:

  • Experience and a strong analytical background, with a degree in computer science, engineering, machine learning, artificial intelligence, deep learning, algorithms, statistics, or a similar field desirable
  • Direct work experience and proficiency in a few of the below technologies and frameworks:
  • Solid experience in building systems with vision processing, deep neural networks, and reinforcement learning
  • Solid understanding of working with Knowledge graphs for AI projects
  • Strong understanding of concepts related to Machine Learning and Statistics
  • Deep understanding of the family of CNN algorithms such as R-CNN, Fast R-CNN, Faster R-CNN, YOLO, etc.
  • Hands-on programming skills in Python
  • Solid hands-on knowledge of using Deep learning libraries such as PyTorch, Fastai, TensorFlow, OpenCV, etc.
  • Proficiency in the implementation of algorithms and the use of advanced data structures to solve problems in computing. A solid foundation in the principles of computer science is desired.
  • Working knowledge of finding ways to quickly experiment and deploy at least 2 DL-based solutions to production
  • Excellent communication and technical skills
  • A can-do attitude and entrepreneurial spirit to conceive ideas, turn challenges into opportunities, and build products
  • Ability to collaborate effectively with highly technical resources in a fast-paced environment
  • Ability to solve complex challenges/problems and rapidly deliver innovative solutions
  • Bachelor’s or Master’s Degree in Computer Science 


  • Hands-on experience with Front end technologies like - HTML, CSS, and JavaScript
  • Publication of DL-based research work in top-tier conferences
  • Previous experience with transformer/vision transformer-based models
  • Understanding distributed computing using Spark 

What sets us apart from the rest

  • Whatfix has been named among the top 20 B2B tech companies like Adobe, PayPal, and Cisco
  • With YoY revenue growth of over 65%, we have also been recognized among the top 20 fastest-growing SaaS companies worldwide in the SaaS 1000 list
  • Recognized by Forrester and Everest Group as a 'Leader' in the digital adoption space, and listed by LinkedIn among one of the Top 5 startups in India in 2020
  • Listed in Deloitte Technology Fast 500™ among fastest-growing companies in North America for 2022 and 2021 and recognized as a Great Place to Work 2022-2023
  • Our Customer centricity is evident from a Customer rating of 4.67 on G2 Crowd & 4.7 on Gartner Peer Insights

Perks / Benefits

  • Equity plan
  • Mac shop, work with the newest technologies
  • Unlimited PTO policy
  • Paid maternity/paternity leave
  • Monthly cell phone stipend
  • Medical, Dental, and Vision coverage (Whatfix pays 80% of the premium for individuals and their families; for the HSA, Whatfix contributes $1,000 for individuals and $2,000 for a family) 
  • Team and company outings
  • Learning and Development benefits

Whatfix is an Equal Opportunity Employer and E-Verify participant. All activities must be in compliance with our Equal Opportunity Laws, ADA, and other regulations, as appropriate.

We value diversity at our company and do not discriminate on the basis of race, religion, color, national origin, gender, sexual orientation, age, marital status, veteran status, or disability status.

Due to the global nature of our company and our hiring committee spanning different time zones, the interviews for this role will be recorded for those not in attendance to review.