Software Engineer - Tools & Automation
Builds and maintains AI-powered diagnostic tools using agentic systems, data pipelines, and ML models. Requires BS degree, 4+ years experience, strong Python, and AI/ML familiarity.
Responsibilities
- Design and implement agentic AI systems with structured interfaces, reasoning loops, and robust error handling
- Build and maintain data pipelines, scheduled workflows, and benchmarking infrastructure
- Develop evaluation and scoring systems to measure and continuously improve model output quality
- Integrate the platform with internal and external services, including ticketing, messaging, storage, and observability
- Collaborate with cross-functional teams to translate business requirements into technical AI solutions
- Architect and maintain production-grade AI solutions with focus on scalability, reliability, and performance
Qualifications
- BS in relevant engineering discipline and 4+ years of relevant work experience
- Strong Python with applications in AI/ML
- Familiarity with AI agents and their applications, including prompt design, tool calling, context management, and evaluation
- Experience or strong interest in Data Science and Engineering
- Experience or strong interest in creating Analytical or AI/ML-based models for real-world problems
- Strong problem-solving skills, especially in simplifying and modelling real-world problems
Bonus Qualifications
- Familiarity with major LLM providers (OpenAI, Anthropic, Google, Meta, etc.) and understanding of their trade-offs in terms of performance, cost, latency, and capabilities
- Experience integrating AI capabilities into applications or eagerness to learn application development
- Full-stack development experience with both backend and frontend technologies
Staff Software Engineer, AI Runtime
Staff Software Engineer building and scaling Databricks' managed large-scale GPU training platform (AIR). Focus on distributed training performance, scheduling, fault tolerance, and developer experience for thousands of accelerators.
Senior Software Engineer, AI Runtime
Senior Software Engineer building and scaling Databricks' managed GPU training platform (AI Runtime) for large-scale distributed AI model training. Requires 5+ years in distributed systems and hands-on experience with GPU training frameworks.
Sr. Machine Learning Engineer, Computer Vision
Build and prototype diffusion-based text-to-image generative models (Pinterest Canvas) using large-scale visual-text datasets. Requires 5+ years industry computer vision experience and an M.S. or Ph.D.
Senior AI/ML Engineer
Senior AI/ML Engineer building transformer and deep learning models on financial and behavioral data to power personalized growth and marketing experiences at Chime. Requires strong production ML experience with PyTorch, AWS, and large-scale data infrastructure.