Senior Software Engineer, AI Frameworks

Job Description

Design and implement end-to-end integrations of Grove with open-source AI frameworks like PyTorch and Ray. Build and maintain production-grade adapters, plugins, and runtime components to ensure scalability and performance across distributed training and inference stacks.

Requirements: Requires a BS/MS/PhD in Computer Science or related field with at least 5 years of experience in AI framework integration. Candidates must possess strong software engineering skills in Go, C++, or Python and practical experience with distributed systems and Kubernetes.

Key Skills: AI Frameworks, PyTorch, Ray, Kubernetes, Go, C++, Python, Distributed Systems, CUDA, GPU Optimization, Model Serving, Containerization, CI/CD, Software Engineering, API Development, Observability

Benefits: Equity, Health Insurance