Software Engineer

Software Engineer

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.

What You'll Do

• 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

What You Bring

• 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

Bonus

• Experience with 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.

Annual base salary range $150,000 - $195,000