The Role
Join a team building and scaling production AI model-serving infrastructure. Collaborate on designing, implementing, optimizing, and monitoring inference solutions; write and review code; test and maintain model-serving systems; and explore new AI/ML techniques to improve performance. Early-career role focused on learning, shipping features, and operational responsibility.
Summary Generated by Built In
About DeepInfra
Why this role matters
What You’ll Do
What You Bring
Bonus
Why DeepInfra
How we work
DeepInfra is building the infrastructure layer for the next generation of AI. We believe open-source models are the future, and companies should have full control over their AI stack without being locked into proprietary providers.
Our inference platform serves trillions of tokens every week across hundreds of production workloads. We build everything from GPU infrastructure to the API layer because every millisecond matters.
We are looking for early-career Software Engineers (0-2 years of experience, including internships) to join our team.
You’ll work closely with our experienced engineers to design, build, and scale infrastructure for serving top open-source AI models. This role is ideal for recent graduates or junior engineers who want to grow quickly while working on high-impact, real production AI systems.
If you’re excited about AI/ML, have taken related courses or built projects, and want to learn how to ship things at scale - we’d love to meet you.
- Collaborate with engineers to 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 (previous exposure helpful - deep expertise not required).
- Monitor and maintain production model-serving systems.
- Work on new features, fix bugs, and contribute to code reviews.
- Participate in daily standups, design reviews, and team discussions.
- Explore new AI/ML techniques and tools, and experiment with improving model performance.
- Try new things. Ship stuff.
What You Bring
- Bachelor’s or Master’s degree in Computer Science, Computer Engineering, or a related field (completed or in final year).
- Strong fundamentals in data structures, algorithms, and software design.
- Proficiency in Python, including experience with AI/ML libraries and frameworks (e.g., NumPy, pandas, SciPy, TensorFlow, PyTorch).
- Experience with AI/ML through coursework, research, personal projects, full-time employment, or internships.
- Familiarity with AI models, Transformers and Diffusers.
- Experience with version control systems (e.g., Git) and agile development methodologies.
- Excellent problem-solving skills, with the ability to debug and optimize code.
- Strong communication and collaboration skills.
- Curiosity, willingness to learn, and desire to build real systems.
- Exposure to C++, CUDA, or AI inference.
- Contributions to open-source ML projects.
Why DeepInfra
- 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.
Three traits define the people who thrive here, and this role leans on all three.
Initiative. We take ownership and step in where we can add value. Whether it’s starting something new, improving what exists, or helping move ideas forward, we aim to be proactive and thoughtful in how we contribute.
Drive. We’re energized by hard problems. Building AI infrastructure is complex, and we lean into that. We care about doing things well, moving fast, and continuously improving — because solving meaningful challenges is what motivates us.
Grit. Things don’t always work on the first try — and that’s expected. We stay persistent, adapt quickly, and learn as we go. We take setbacks seriously, but not personally, and use them to get better.
Skills Required
- Bachelor's or Master's degree in Computer Science, Computer Engineering, or related field (completed or in final year).
- Strong fundamentals in data structures, algorithms, and software design.
- Proficiency in Python, including experience with AI/ML libraries and frameworks (NumPy, pandas, SciPy, TensorFlow, PyTorch).
- Experience with AI/ML through coursework, research, personal projects, full-time employment, or internships.
- Familiarity with AI models, including Transformers and Diffusers.
- Experience with version control systems (e.g., Git) and agile development methodologies.
- Excellent problem-solving skills, with the ability to debug and optimize code.
- Strong communication and collaboration skills.
- Curiosity, willingness to learn, and desire to build real systems.
- Exposure to C++, CUDA, or NCCL (helpful but deep expertise not required).
- Contributions to open-source ML projects.
- Exposure to AI inference engineering.
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.
