DeepInfra seeks early-career Software Engineers to design and scale infrastructure for AI models. Responsibilities include collaboration on AI solutions, coding, testing, and maintaining production systems, alongside learning and growth opportunities.
DeepInfra is 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.
Similar Jobs
Artificial Intelligence • Machine Learning • Natural Language Processing • Software • Conversational AI
The Pre-Sales Solutions Engineer leads technical discovery, designs and executes POCs, advises on architectural needs, triages issues, and collaborates with teams to ensure customer success and continuous engagement throughout the project lifecycle.
Top Skills:
DockerHTTPJavaScriptKubernetesPythonSipTypescriptWebrtc
Artificial Intelligence • Machine Learning • Natural Language Processing • Software • Conversational AI
The Solutions Architect will lead enterprise deployments, ensure technical success for customers, and contribute to product improvements. Responsibilities include architecture design, post-sales support, technical problem-solving, and customer engagement.
Top Skills:
DockerJavaScriptKubernetesPythonRust
Artificial Intelligence • Machine Learning • Software
Assist the engineering team in designing, developing, and deploying scalable AI models. Responsibilities include implementing models, optimizing code, and participating in collaboration activities.
Top Skills:
C++CudaGitNcclNumpyPandasPythonPyTorchScipyTensorFlow
What you need to know about the NYC Tech Scene
As the undisputed financial capital of the world, New York City is an epicenter of startup funding activity. The city has a thriving fintech scene and is a major player in verticals ranging from AI to biotech, cybersecurity and digital media. It also has universities like NYU, Columbia and Cornell Tech attracting students and researchers from across the globe, providing the ecosystem with a constant influx of world-class talent. And its East Coast location and three international airports make it a perfect spot for European companies establishing a foothold in the United States.
Key Facts About NYC Tech
- Number of Tech Workers: 549,200; 6% of overall workforce (2024 CompTIA survey)
- Major Tech Employers: Capgemini, Bloomberg, IBM, Spotify
- Key Industries: Artificial intelligence, Fintech
- Funding Landscape: $25.5 billion in venture capital funding in 2024 (Pitchbook)
- Notable Investors: Greycroft, Thrive Capital, Union Square Ventures, FirstMark Capital, Tiger Global Management, Tribeca Venture Partners, Insight Partners, Two Sigma Ventures
- Research Centers and Universities: Columbia University, New York University, Fordham University, CUNY, AI Now Institute, Flatiron Institute, C.N. Yang Institute for Theoretical Physics, NASA Space Radiation Laboratory
