Summary
Join a fintech platform backed by industry leaders to create engaging wealth experiences through AI and investment intelligence powered personalization. We are seeking a Staff LLM Data Scientist for Sage with Large Language Model (LLM) experience who can work end-to-end, training, and productionalizing models.
Requirements
- Ph.D./Masters/Bachelor's degree in computer science, mathematics, statistics, engineering, or relevant field
- Experienced in the field of NLP/LLM and well-versed with the current and latest state-of-the-art research
- Hands-on experience in various LLM fine-tuning techniques (e.g. LORA), LLM inference frameworks (e.g. vLLM), advanced RAG pipelines
- Excellent knowledge of LLM evaluation methods and metrics
- 5-6+ years of machine learning/deep learning experience within frameworks such as TensorFlow and/or PyTorch
- 2+ years of practical experience in the development of generative AI applications
- Publications at reputable machine learning conference or journal
- Proficient in Python and SQL
- Analytical and problem-solving skills
- Ability to visualize data in the most effective way possible for a given project or study
- Thrives in a highly demanding, entrepreneurial, and fast-paced environment
- Is a top performer and has a proactive, 'doer', and problem-solver mentality
- Is highly flexible, has a good tolerance for ambiguity, and can quickly adapt to changing priorities
- Is an exceptional team player with solid communication skills
Responsibilities
- Design and fine-tune open source and proprietary LLMs for various tasks such as answering questions, summarization, reasoning and planning, etc
- Build advanced Retrieval Augmented Generation (RAG) pipeline including rewriting, embedding fine-tuning, hybrid search, reranking, knowledge graphs, etc
- Implement a comprehensive evaluation framework and metrics for model performance
- Deploy models into production environments and ensure low latency, reliability, and scalability
- Collaborate with product team and software engineering team to build end-to-end product systems
Benefits
- Performance linked variable compensation, including equity
- Medical, dental, vision, life and disability insurance
- Flexible Spending Account (FSA) and Health Savings Account (HSA)
- 401(k) Retirement Plan
- Flexible PTO policy and Company-paid holidays
- Parental Leave: 12 week paid maternity, 6 week paid paternity leave
- Corporate Social Responsibility and volunteering opportunities
- Access to our Chief Mindfulness Officer for Mindfulness events and 1:1 personal coaching
- Company sponsored events like Qi Gong, mindfulness events, development workshops, team dinners, pickleball games, happy hours, holiday parties and more!