Demo

Lead AI/ML Engineer

ProSearch
San Francisco, CA Full Time
POSTED ON 2/22/2025
AVAILABLE BEFORE 5/10/2025

As an AI / ML Engineer, you will be responsible for the design and implementation of the cloud-native AI / ML framework and infrastructure (supporting both GenAI and traditional AI / ML technology and applications), AI model productionalization (E2E AI / ML model production lifecycle : AI / ML model development, deployment, monitoring, refresh, etc), Data Lakehouse, and related technology, systems, and tools. This technologist will drive the technology development and architecture in these areas to ensure the platforms, systems, tools, models, and services generated from this team are technically sound with strong security, quality, reliability, usability, scalability, performance, efficiency, and operability to serve the fast-growing needs of our client and their customers. It’s important to balance the near term and long-term needs.

This is a fantastic opportunity for a highly motivated and passionate software engineer to apply deep expertise and rich experience in Data Lake and AI / ML Engineering and related areas to develop technology, platforms, and services, and apply them to multiple lines of business with a big customer base to generate real business impact in a fast growing commerce platform and payment tech company during its technology transformation stage to meet high customer demands. At the same time, you will learn and grow your career by solving large scale and complex problems.

This AI / ML Engineer will report to the Senior Director of AI / ML Engineering located in the San Francisco Bay Area. The ideal candidate will be a hands-on leader with deep technical expertise in building and scaling data lake environments, coupled with strong people leadership and communication skills to drive alignment across the organization.

How you’ll make an impact

  • Contribute to the development of technology and platform for the company's AI / ML needs, ensure the functional richness, reliability, performance, and flexibility of this platform
  • Help with the design, architecture and implementation of the AI / ML infrastructure, platform and services.
  • Help design and implement the choice for our client’s AI Orchestration framework.
  • Help design and implement the choice for our client’s Vector database along with integrating embedding algorithms in the Data Lakehouse.
  • Challenge the status quo and hold a high bar to drive for better and better technical solutions to effectively serve all the needs in the broad risk area of Wex.
  • Closely partner with the other senior engineers across the organizations at Wex to deeply understand the business needs and jointly come up with technical solutions to serve these needs effectively. Demonstrate strong thought leadership and partnership.
  • Partner closely with information security teams to ensure the security and privacy of data, implementing encryption and access controls as required.
  • Partner with solutions architecture and LOB product and engineering orgs on integrations of the AI / ML Platform with their products / systems, and ensure reliable and performed execution.
  • Engage with engineers in the team to evaluate technologies and services that complement the company’s AI / ML strategy.
  • Report on the status of development, quality, operations, and system performance to management and partners.

Qualifications & Experience

  • Bachelor's degree in Computer Science, Software Engineering, or a related field. OR demonstrable equivalent deep understanding, experience, and capability.
  • A Master's or PhD degree in Computer Science is a plus. A technically deep and innovative Software Engineer leader able to act and deliver to the business needs.
  • 5 years of ML engineering, software design, development experience at a large scale.
  • Strong software development skills in his or her chosen programming language, but ready and willing to learn and program in Java and Python.
  • Broad understanding of distributed computing frameworks like Ray and Spark
  • Demonstrable understanding of and hands-on experience in supervised and unsupervised ML and AI
  • Demonstrable understanding of and hands-on experience in building Retrieval Augmented Generation applications, from AI orchestration frameworks and embeddings to vector databases.
  • Demonstrable experience in Kubernetes and Terraform
  • Experience with Sagemaker a plus
  • Broad understanding of cloud architecture tools and services, such as S3, EMR, Kubernetes, Lambda functions. Experience in AWS and Azure is highly desirable.
  • Deep technically; hands-on in architecture design, development, deployment, and production operation; familiar with agile software development and modern development tools and frameworks; holding strong engineering principles including automation and quality as well as best practices with a high bar. Extensive experience in full software development life cycle E2E including in production monitoring.
  • Rich experience and deep expertise in the areas of Big Data and large scale data platforms, especially in AI and ML.
  • Highly experienced and deep understanding of CI / CD automation.
  • Rich experience in combined engineering practice and agile development. Experienced in leading and guiding teams for effective development using this approach.
  • Excellent analytical skills; Capability in identifying and resolving complex problems.
  • An understanding of and experience in payment organizations is a plus.
  • If your compensation planning software is too rigid to deploy winning incentive strategies, it’s time to find an adaptable solution. Compensation Planning
    Enhance your organization's compensation strategy with salary data sets that HR and team managers can use to pay your staff right. Surveys & Data Sets

    What is the career path for a Lead AI/ML Engineer?

    Sign up to receive alerts about other jobs on the Lead AI/ML Engineer career path by checking the boxes next to the positions that interest you.
    Income Estimation: 
    $119,030 - $151,900
    Income Estimation: 
    $149,493 - $192,976
    Income Estimation: 
    $149,493 - $192,976
    Income Estimation: 
    $184,796 - $233,226
    Income Estimation: 
    $184,796 - $233,226
    Income Estimation: 
    $179,606 - $233,815
    Income Estimation: 
    $101,387 - $124,118
    Income Estimation: 
    $119,030 - $151,900
    Income Estimation: 
    $119,030 - $151,900
    Income Estimation: 
    $149,493 - $192,976
    View Core, Job Family, and Industry Job Skills and Competency Data for more than 15,000 Job Titles Skills Library

    Job openings at ProSearch

    ProSearch
    Hired Organization Address Westbrook, ME Full Time
    Direct Hire Westbrook, Maine Location: Westbrook, Maine Type: Direct Hire Job #18374 Management Trainee – Electrical Dis...
    ProSearch
    Hired Organization Address Portland, ME Temporary
    Temporary portland Location: portland Type: Temporary Job #19972 Portland medical group is seeking a temporary Billing S...
    ProSearch
    Hired Organization Address Portland, ME Full Time
    About the Company - Looking for a fun and professional work environment? Look no further! Our client, a busy boutique pr...
    ProSearch
    Hired Organization Address Bath, ME Full Time
    Our client, an international IT services provider, is seeking to add a Senior Linux Administrator/Engineer to their grow...

    Not the job you're looking for? Here are some other Lead AI/ML Engineer jobs in the San Francisco, CA area that may be a better fit.

    Lead AI/ML Engineer

    Pro Search, San Francisco, CA

    ML Engineer

    Navier AI, San Francisco, CA

    AI Assistant is available now!

    Feel free to start your new journey!