Software Engineer, Backend & ML Infra
170k - 200k USD
Remote
Full Time
#Engineering
#AI
#Machine Learning
#Python
#Kubernetes
#Dask
#PostgreSQL
#MLFlow
#Databases
#Data Management
Are you ready to help us make artificial intelligence accessible to everyone within every organization? At Graft, we are building an intelligence layer that bridges the gap between company knowledge and the people who need it most. Our platform serves as a central hub where predictive, generative, and search AI come together to help teams unlock the full potential of their enterprise data. We are proud to be supported by prominent investors like GV, Radical Ventures, NEA, SV Angel, Essence VC, and Formulate Ventures, and we invite you to help us shape the future of how teams interact with collective intelligence.
What you bring
We are looking for a mid-level software engineer to join our team in a full-time, remote capacity. You will play a vital role in our engineering department, focusing on backend systems and machine learning infrastructure. Your work will directly impact how our users interact with complex data. You will be based anywhere, provided you are located in the United States.
- You will design, implement, and optimize the workflow orchestration and scheduling systems that power our platform.
- You will build and manage production-grade pipelines for various machine learning applications, such as chat systems, classification, and semantic search.
- You will tackle complex challenges related to the scalability of our compute resources and artifact storage.
Requirements
To succeed in this role, you should have at least three years of experience as a backend engineer and a strong command of Python. We look for candidates who have hands-on technical experience with Kubernetes, Dask, and PostgreSQL. Familiarity with machine learning tools like MLFlow, vector databases, and Sematic is essential. You should possess a solid grasp of machine learning fundamentals, including embeddings, foundation models, and large language models. We value individuals who are excellent communicators, thrive in fast-paced startup environments, and have a genuine passion for solving real-world problems through technology. Proficiency in English is required.
What we offer
We offer a competitive salary range of $170,000 - $200,000 USD. In addition to your compensation, we provide the following benefit:
- Remote work flexibility.
How to Apply
If you are passionate about building systems at scale and want to expand the impact of machine learning, we would love to hear from you. Please reach out to our team to start a conversation about how your skills and experience can help us achieve our mission.





