Staff Machine Learning Engineer, Applications

Job Description

Job Description

Who We Are:

SmithRx is a rapidly growing, venture-backed Health-Tech company. Our mission is to disrupt the expensive and inefficient Pharmacy Benefit Management (PBM) sector by building a next-generation drug acquisition platform driven by cutting edge technology, innovative cost saving tools, and best-in-class customer service. With hundreds of thousands of members onboarded since 2016, SmithRx has a solution that is resonating with clients all across the country.

We pride ourselves for our mission-driven and collaborative culture that inspires our employees to do their best work. We believe that the U.S healthcare system is in need of transformation, and we come to work each day dedicated to making that change a reality. At our core, we are guided by our company values:

• Integrity: Our purpose guides our actions and gives us confidence in the path ahead. With unwavering honesty and dependability, we embrace the pressure of challenging the old and exemplify ethical leadership to create the new.
• Courage: We face continuous challenges with grit and resilience. We embrace the discomfort of the unknown by balancing autonomy with empathy, and ownership with vulnerability. We boldly challenge the status quo to keep moving forward—always.
• Together: The success of SmithRx reflects the strength of our partnerships and the commitment of our team. Our shared values bind us together and make us one. When one falls, we all fall; when one rises, we all rise.

Job Summary:

SmithRx is leading the transformation of pharmacy benefit management (PBM) with a cutting-edge platform that delivers real-time insights, cost efficiencies, and exceptional customer experiences. We are seeking an experienced Data Scientist with expertise in generative AI as we continue to expand. In this key role, you will drive AI innovation and lead the implementation of AI-powered solutions. The position requires at least 5 years of experience in machine learning and AI development, with proven success leading AI initiatives. You will work with cross-functional leaders to deliver impactful AI solutions that directly influence our business outcomes.

What you will do:

As a member of the SmithRx Application Engineering and AI team, you will work customer facing as well as internal tooling that leverages AI to improve productivity and quality. You will work directly with software engineers, data engineers, product managers, and subject matter experts in the definition of new generative AI capabilities able to provide our members and internal organizations with the information they need to make proper decisions.

Qualified candidates will have an in-depth knowledge of the most common Large Language Models (LLMs) and Retrieval Methods, know how to train and fine-tune LLMs, and design and implement LLM-powered agents and tools at scale.

Responsibilities

• Design Generative AI solutions, such as information retrieval and summary generation for support and operations organizations.
• Partner with Software Engineering teams to build and deploy GenAI applications.
• Ideate and explore opportunities to deploy GenAI technologies for customer-facing experiences
• Drive the adoption and scalability of Generative AI capabilities within SmithRx; advocacy on the technology and evangelize the use of GenAI.

What you will bring to SmithRx:

• 5+ years of experience in data science, machine learning, and AI development, with proven success leading AI initiatives
• MS or PhD in Computer Science, Electrical Engineering, Statistics, Robotics or equivalent fields.
• Applied Machine Learning experience (regression and classification, supervised, and unsupervised learning).
• Strong mathematical background (linear algebra, calculus, probability, and statistics).
• Proficiency in Python and object-oriented programming.
• Strong experience working with machine learning and natural language processing techniques and tools.
• Strong experience using Generative AI models, with a good understanding of deep learning model classes such as GPT, VAE, and GANs, as well as their hyperparameters.
• Strong experience with retrieval methods e.g. using embeddings.
• Strong experience using key Python packages for data wrangling, machine learning and deep learning such as pandas, sklearn, TensorFlow, torch, transformers, LangChain, etc.
• Experience in Prompt Engineering and few-shot techniques to enhance LLM's performance on specific tasks.
• Experience with training and fine-tuning deep learning models, especially LLMs, and how to tune hyperparameters to ensure task generalization.
• Ability to drive a project and work both independently and within a cross-functional team.
• Excellent verbal and written communication, able to articulate complex concepts with a non-technical audience.

Preferred Qualifications

• Experience with embedding model training and retrieval method evaluation approaches.
• Experience with LLM architectures, adapters, Mixture of Experts (MoEs) pretraining and fine-tuning techniques.
• Experience with design, deployment, and evaluation of LLM-powered agents and tools and orchestration approaches.
• Experience with reinforcement learning approaches in the context of fine-tuning LLM outputs.
• Experience with time series analysis and multivariate time series modeling.

What SmithRx Offers You:

• Highly competitive wellness benefits including Medical, Pharmacy, Dental, Vision, and Life Insurance and AD&D Insurance
• Flexible Spending Benefits
• 401(k) Retirement Savings Program
• Short-term and long-term disability
• Discretionary Paid Time Off
• Paid Company Holidays
• Wellness Benefits
• Commuter Benefits
• Paid Parental Leave benefits
• Employee Assistance Program (EAP)
• Well-stocked kitchen in office locations
• Professional development and training opportunities

Location: US - Remote