Design and develop inference solutions for AI models, optimize performance, and maintain production systems while owning end-to-end problems.
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 strong 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.
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.
CompensationThe base pay range for this role is $150,000 – $195,000 per year.
Deep Infra Inc. Palo Alto, California, USA Office
Palo Alto, California, United States, 94306
Similar Jobs
Cloud • Computer Vision • Information Technology • Sales • Security • Cybersecurity
Develop and maintain CrowdStrike’s macOS endpoint security sensor using C/C++ and Swift. Design, implement, and own features across operating-system subsystems including filesystems, memory, processes, and networking. Build reliable, high-performance, multithreaded software, troubleshoot customer-reported issues, collaborate with distributed teams, and use AI technologies to improve workflows and decision-making.
Top Skills:
CC++CybersecurityEndpoint ProtectionLinuxmacOSMultithreadingNetworking ProtocolsSwiftWindows
Cloud • Computer Vision • Information Technology • Sales • Security • Cybersecurity
Build and maintain sensor quality dashboards, release-health metrics, developer tooling, and AI-assisted workflows. The role uses Python and Django on the backend, React and CSS on the frontend, and works with LogScale, PostgreSQL, and Valkey to collect and surface sensor health data. Responsibilities include defining health signals, participating in repo-health rotations, improving engineering visibility, and developing Claude Code skills and agentic workflows.
Top Skills:
Agentic WorkflowsAi-Assisted DevelopmentClaude CodeCSSDjangoGitlab Ci/CdLogscalePostgresPythonReactValkey
Artificial Intelligence • Cloud • Information Technology • Consulting
Develops and executes functional, scale, performance, and regression tests for networking products. Responsibilities include creating test plans, automating test cases with Python, Go, or Perl, debugging issues with software development teams, reporting defects, integrating automation frameworks, and validating L2/L3 networking technologies across multiple platforms. The role requires strong programming, networking, operating-system, software architecture, and test methodology expertise.
Top Skills:
BgpContainerizationEthernetFreebsdGoIp MulticastIs-IsIxiaJSONL2VpnL3VpnLagLinuxMc-LagMplsNetwork VirtualizationOspfPerlPythonRobot FrameworkRsvpSnmpSoftware-Defined Networking (Sdn)SpirentStpTcp/IpTelemetryVlanVMwareVplsYaml
What you need to know about the San Francisco Tech Scene
San Francisco and the surrounding Bay Area attracts more startup funding than any other region in the world. Home to Stanford University and UC Berkeley, leading VC firms and several of the world’s most valuable companies, the Bay Area is the place to go for anyone looking to make it big in the tech industry. That said, San Francisco has a lot to offer beyond technology thanks to a thriving art and music scene, excellent food and a short drive to several of the country’s most beautiful recreational areas.
Key Facts About San Francisco Tech
- Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
- Major Tech Employers: Google, Apple, Salesforce, Meta
- Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
- Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
- Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
- Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine
