Job Description

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.

Job Tags

Similar Jobs