SoFi
Senior Staff AI Engineer
CA - San Francisco · Staff+
Sponsorship not specifiedDetected 7 days ago
PythonReactElasticsearchSnowflakeVector DatabasesAWSGCPAzureCloud PlatformsDockerKubernetesMachine LearningData EngineeringData ScienceNLPLLMsRAGLangGraphAI OrchestrationComplianceCustomer SupportCollaborationProblem SolvingMentoring
About the role
- SoFi's Senior Staff AI Engineer is a hands-on AI engineering role in SoFi's growing independent risk organization.
- This is a critical, senior role responsible for setting the technical direction, driving execution, and ensuring the successful delivery of our most complex, production-level AI initiatives.
- This hands-on role will work closely with the Director of Risk Analytics, and will leverage your deep expertise to solve our hardest problems, mentor the next generation of engineers, and directly connect technical innovation to major business success.
Responsibilities
- Distributed Agent Memory & State: Develop robust, persistent infrastructure for agentic state management, ensuring that long-running agent workflows maintain context and reliability across distributed nodes and regional failovers
- Model Serving Infrastructure: Support the development of a unified model serving platform designed to host internally fine-tuned and custom-trained models to ensure high-throughput, low-latency inference across diverse hardware footprints.
- Develop robust, persistent infrastructure for agentic state management, ensuring that long-running agent workflows maintain context and reliability across distributed nodes and regional failovers
- Support the development of a unified model serving platform designed to host internally fine-tuned and custom-trained models to ensure high-throughput, low-latency inference across diverse hardware footprints.
Requirements
- Expert-level experience with LangGraph to model and orchestrate complex, stateful multi-step reasoning and control flow in LLM applications.
- Expert-level proficiency in developing sophisticated agentic solutions, with a portfolio demonstrating advanced use of planning, memory management, tool integration, and control flow.
- Deep experience designing and institutionalizing AI observability solutions (e.g., LangSmith, Arize, Deepchecks) and advanced tracing and testing methodologies for LLM and agentic systems.
- Experience with cloud platforms (AWS, Azure, or GCP) and containerization technologies (Docker, Kubernetes).
- Expert level Python is required.
- Experience with large-scale data handling, including unstructured and structured data pipelines, with a strong preference for Snowflake and DynamoDB.
- Experience with vector databases and retrieval-augmented generation (RAG) techniques using systems like Elasticsearch, Pinecone, or FAISS for enhancing LLM performance.
- Strong analytical and problem-solving skills with attention to detail and an ability to work with complex, large-scale systems.
Nice to have
- Familiarity with regulatory frameworks and ethical considerations in AI within the banking industry (e.g., GDPR, data privacy, model explainability).
- The Company hires the best qualified candidate for the job, without regard to protected characteristics.
- Pursuant to the San Francisco Fair Chance Ordinance, we will consider for employment qualified applicants with arrest and conviction records.
- New York applicants: Notice of Employee Rights
- SoFi is committed to an inclusive culture.
- Internal Employees
- If you are a current employee, do not apply here - please navigate to our Internal Job Board in Greenhouse to apply to our open roles.
- React is strongly preferred.
Compensation
- The base pay range for this role is listed below.
- Final base pay offer will be determined based on individual factors such as the candidate's experience, skills, and location.
Benefits
- Bachelor's or Master's degree in Computer Science, Data Science, AI, Machine Learning, or a related field.
- To view all of our comprehensive and competitive benefits, visit our Benefits at SoFi page!
Company info
- Shape a brighter financial future with us.
- Together with our members, we're changing the way people think about and interact with personal finance.
- We're a next-generation financial services company and national bank using innovative, mobile-first technology to help our millions of members reach their goals.
- The industry is going through an unprecedented transformation, and we're at the forefront.
- We're proud to come to work every day knowing that what we do has a direct impact on people's lives, with our core values guiding us every step of the way.
- Join us to invest in yourself, your career, and the financial world.
- This role will be instrumental in conceptualizing, prototyping and implementing best-in-class AI-based solutions to meet risk management and compliance requirements.
- This is a crucial role for the independent risk function as we execute our mission to help more members get their money right.
- Architecture and Strategy: Define the long-term technical architecture and strategy for our next-generation AI platform, particularly focusing on robust, scalable agentic frameworks and LLM deployment patterns.
- Advanced LLM Orchestration: Architect and standardize the use of graph-based LLM orchestration, leveraging expert-level mastery of LangGraph to solve highly complex, multi-stage reasoning problems at scale.
This listing is sourced directly from SoFi's careers page and normalized into a canonical job model.