Principal Pytorch Staff Engineer
- Red Hat
- 8 - 13 years
- Bengaluru
- 5 months ago
- Email to a friend
- Report this job
Job Description
Job Summary:
At Red Hat, we connect an innovative community of customers, partners, and contributors to deliver an open-source stack of trusted, high-performing solutions. We offer cloud, Linux, middleware, virtualization, and AI technologies, together with award-winning global customer support, consulting, and implementation services. Red Hat is a rapidly growing company supporting more than 90% of Fortune 500 companies.
The AI Core Components organization, part of AI Engineering, is looking for individuals with a passion for Open Source, automation and helping grow the impact of Red Hats AI offerings for our customers and the community.
As a Staff Engineer youd be a person with deep expertise in Machine Learning systems and tools, relentless curiosity, and a bias for action. This role is ideal for someone who thrives at the intersection of AI innovation and engineering excellence and is passionate about boosting engineering productivity through the intelligent use of the latest AI advancements.
What you will do:
- Prototype with Purpose: Rapidly build PoCs using the latest AI models and frameworks to enhance developer workflows across the PyTorch engineering lifecycle (code, test, document, debug, optimize).
- Bridge Industry & Implementation: Evaluate innovations from open source, academia, and industry and assess how they can improve the way PyTorch is built and maintained.
- Drive Developer Productivity: Create intelligent tools that support faster iteration, better testing, and smoother onboarding for PyTorch engineersturning AI into leverage.
- Embed in the Team: Work alongside core PyTorch engineers to understand friction points and infuse AI-powered solutions into existing workflows and CI pipelines.
- Lead by Influence: Set technical direction in the space of AI-enhanced engineering productivity. Act as a multiplier for the teammentoring, demoing, writing internal blogs, and making innovation accessible.
- Contribute to the PyTorch Ecosystem: Your work will extend to open-source tooling or enhancements that benefit the broader PyTorch community.
- 8+ years of experience in machine learning, software engineering, or systems design, with significant exposure to AI/ML infrastructure.
- Deep understanding of LLMs, AI agents, vector search, prompt engineering, and developer productivity tools.
- Strong Python programming skills and experience with PyTorch, Hugging Face Transformers, LangChain, vLLM, or OpenAI/Anthropic APIs.
- Demonstrated ability to build fast, impactful prototypes that lead to production-ready tools.
- Experience improving engineering workflows through AI toolscode generation, automated test generation, bug triage, documentation assistants, etc.
- Familiarity with modern software engineering practices: CI/CD, DevOps, VS Code extensions, or internal tooling development.
- Collaborative mindset and excellent communication skillscomfortable working with developers, QE, product managers, and leadership.
- Contributions to PyTorch or other open-source ML/AI frameworks.
- Experience building developer tools or productivity platforms within ML orgs.
- Familiarity with AI use cases in systems optimization, model compilation, or hardware acceleration.
- Exposure to long-context transformer models and techniques like EasyContext, NoLiMa, or KV-cache paging.
Job Classification
Industry: IT Services & ConsultingFunctional Area / Department: Engineering - Software & QARole Category: Software DevelopmentRole: Technical ArchitectEmployement Type: Full timeContact Details:
Company: Red HatLocation(s): Bengaluru
Keyskills: python machine learning artificial intelligence productivity vector continuous integration kubernetes uds canoe system design photoshop canalyzer autosar lin capl devops linux embedded c debugging software engineering can bus illustrator
Similar positions
Red Hat
Founded in 1993, Red Hat is the premier Linux and open source provider. The most recognized Linux brand in the world. We serve global enterprises through technology and services made possible by the open source model. Solutions include Red Hat Enterprise Linux operating platforms, sold through a sub...
Job Listings
We are in open beta version of website, please let us know if any issues, at: info(at)indigojobs.in