The Role
Design and develop inference solutions for AI models, optimize performance, and maintain production systems while owning end-to-end problems.
Summary Generated by Built In
DeepInfra is looking for early-career Software Engineers to join our team. You’ll work on designing, building, and scaling infrastructure for serving top open-source AI models in production. This role is ideal for engineers who are already comfortable owning problems end-to-end and want to deepen their experience working on high-impact AI systems.
If you’re excited about AI/ML, have built and shipped projects, and are looking to work on real systems at scale — we’d love to meet you.
- Design, develop, and test inference solutions for state-of-the-art AI models
- Implement, optimize, and evaluate AI models using Python, C++, CUDA, and NCCL
- Own and operate production model-serving systems, including monitoring and debugging
- Build new features, improve system performance, and contribute to overall system design
- Participate in code reviews and technical discussions to maintain high engineering standards
- Explore and apply new AI/ML techniques to improve model performance and efficiency
- Take ideas from concept to production
- Bachelor’s or Master’s degree in Computer Science, Computer Engineering, or a related field
- 1–4 years of relevant experience, including early full-time roles or research
- Strong fundamentals in data structures, algorithms, and software design
- Proficiency in Python and experience working with AI/ML frameworks (e.g., PyTorch, TensorFlow)
- Hands-on experience building, shipping, and maintaining software systems
- Familiarity with AI models, Transformers, and Diffusers
- Experience working with version control (Git) and collaborative development workflows
- Ability to debug, optimize, and improve existing systems
- Strong communication skills and ability to work independently in a fast-paced environment
- Experience with C++, CUDA, or AI inference
- Contributions to open-source ML projects
- Work on cutting-edge AI model serving - the systems that power the next generation of LLMs and multimodal models.
- Small team, huge impact: your work ships directly to customers.
- Opportunity to learn from engineers building high-performance inference at scale.
- Fast-paced environment with ownership, autonomy, and end-to-end responsibility.
Annual base salary range
$150,000 - $195,000
Skills Required
- Bachelor's or Master's degree in Computer Science, Computer Engineering, or a related field
- 1 - 4 years of relevant experience
- Proficiency in Python
- Experience working with AI/ML frameworks (e.g., PyTorch, TensorFlow)
- Strong fundamentals in data structures, algorithms, and software design
- Hands-on experience building, shipping, and maintaining software systems
- Familiarity with AI models, Transformers, and Diffusers
- Experience with version control (Git)
- Strong communication skills and ability to work independently in a fast-paced environment
Am I A Good Fit?
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.
Success! Refresh the page to see how your skills align with this role.
The Company
What We Do
Let Deep Infra run your ML infrastructure. Just use our top AI models using a simple API or deploy your own model with us.
