Senior/Principal Applied Data Scientist

About the Role

This is a contract position expected to last 3 months, with the possibility of extension for additional months or even a full time position. Expected commitment is full-time when on the contract.

About Cake

Cake is on a mission to make cutting-edge AI accessible to every business. Backed by Google’s Gradient Ventures, our platform simplifies AI adoption with a full-stack approach, from compute infrastructure to production-ready workflows, at 10x the speed and cost-efficiency of alternatives. We're scaling rapidly across mid-market enterprises, and we're looking for a dynamic, solutions-focused Data Scientist to help power this growth.

Intensity, humility, and collaboration are the cornerstones of our culture. We’re a dynamic team of forward-thinking technologists who love to learn from each other and push the boundaries of AI at an incredibly fast pace. If you’re excited about reshaping the future of AI, we’d love to hear from you.

About the Role

We’re looking for a Senior/Principal Data Scientist who thrives at the intersection of AI research, customer success, and engineering. You’ll help design intelligent agentic workflows, build novel document extraction pipelines, and support customers in deploying production-grade AI systems.

What You’ll Do

• Data science and solutions engineering to build and design agentic workflows
• Solve novel, complex data science problems in document classification and structured data extraction
• Write code to extract structured information from unstructured patient records
• Leverage deep technical expertise to support customers and troubleshoot complex issues
• Develop and expand a robust customer knowledge base to answer common questions, significantly reducing onboarding time for new customers.
• Work with the product team to prioritize the roadmap based on customer feedback
• Communicate findings to stakeholders in a clear and concise way
• Ensure privacy and Compliance standards (HIPAA, PHI/PII) are upheld throughout the ML lifecycle

About You

• Background in data science, agentic frameworks, machine learning, and engineering
• Strong programming skills in Python
• Deep experience extracting structured data from unstructured PDFs using frameworks like: DSPy, Pydantic, Trustcall, LangGraph, LangChain, LlamaIndex, and Docling
• Deep understanding of LLM models
• Work closely with subject matter experts to create and refine evaluations sets
• Experience working with both technical and non-technical stakeholders
• Excellent communication, collaboration, and customer relationship management skills
• Strong grasp of the full MLOps lifecycle, including data engineering, training/evaluation, serving, and monitoring
• Proficiency in creating Docker containers

Nice to Haves

• Experience with cloud platforms (AWS, GCP, Azure) and ideally Kubernetes
• Experience in Insurance and Healthcare Industry
• Familiarity with Bedrock and Anthropic models
• Experience working as part of a startup or lean team

Location

The role is remote-friendly. Expectations are that you will work and be available in EST timezone.

Compensation

• $12,000–$22,500 per month for the 3-month engagement, commensurate with experience and qualifications
• We’re happy to discuss range expectations early in the process

Equal Opportunity Employer

Cake is committed to providing equal employment opportunities to all employees and job seekers regardless of race, color, religion, national origin, sexual orientation, gender, gender identity, marital status, disability, veteran status, or any other legally protected category. As an equal opportunities employer, we value diversity and its positive impact on our culture.

Cake also complies with the Americans with Disabilities Act (ADA). We are dedicated to working with and providing reasonable accommodation to job applicants with physical or mental disabilities. If you require accommodation, please email us at accommodationrequest@cake.ai and we will promptly address your request.