Software Engineer, Backend and ML Infra

Summary

Join Graft, a company building an AI-powered knowledge layer for organizations, as a Backend Engineer. This remote, US-based role focuses on building and maintaining production pipelines for diverse machine learning applications. You will design, implement, and optimize workflow orchestration and scheduling systems, ensuring scalability for both compute and artifact storage. The ideal candidate possesses at least 3 years of backend engineering experience, proficiency in Python, and hands-on experience with Kubernetes, Dask, and PostgreSQL. A solid understanding of machine learning fundamentals is crucial. Graft offers a competitive salary and a collaborative work environment.

Requirements

• A minimum of 3 years of experience as a backend engineer
• Proficiency in Python programming
• Hands-on experience with Kubernetes, Dask, and PostgreSQL
• Familiarity with ML-specific open-source tools such as MLFlow, Sematic, and vector databases
• Solid understanding of machine learning fundamentals, particularly foundation models, embeddings, and large language models (LLMs), and a passion for applying ML to solve real-world problems
• Strong problem-solving abilities and a track record of thriving in fast-paced startup environments
• Excellent communication skills and a demonstrated ability to collaborate effectively within a team

Responsibilities

• Build and maintain production pipelines for diverse Machine Learning applications, including semantic search, classification, and chat systems
• Design, implement, and optimize workflow orchestration and scheduling systems
• Work on scalability, for both compute and artifact storages
• Implement advanced techniques for processing and managing unstructured data
• Help implement techniques for processing unstructured data
• Contribute to product strategy with a focus on usability for non-expert users

Benefits

• Remote work, flexible hours
• Estimated Salary Range: $170,000 - $200,000