Distinguished Engineer, JAX
Location
California
Posted
147 days ago
Salary
$320K - $488.8K / year
Seniority
Lead
Job Description
Distinguished Engineer, JAX
NVIDIA
• Play significant role in NVIDIA's effort in contributing to JAX. • Design and implement JAX core components and drive peak performance on NVIDIA products. • Work with AI applied researchers and leaders to build future-proof models. • Build tools that will increase the efficiency of teams developing AI-based systems. • Work to bridge the gap between the latest in numerical computing, simulation and deep learning research and their applications in real world products.
Job Requirements
- BS in Computer Science or Computer Engineering or related field (or equivalent experience)
- 18+ years relevant experience
- C/C++ and Python programming
- Experience with machine learning frameworks and their internals (e.g. PyTorch, TensorFlow, scikit-learn, etc.)
- Proven ability developing customer-facing solutions, balancing feature requests and bugs.
- Proven technical foundation in CPU and GPU architectures, numeric libraries, modular software design.
- Highly motivated with excellent verbal and written communication skills.
- Ability to work successfully with multi-functional teams, principles and architects.
- Coordinates effectively across organizational boundaries and geographies.
Benefits
- equity
- benefits
Related Guides
Related Categories
Related Job Pages
More Engineer Jobs
Senior Systems and Certification Engineer
Collinear GroupEngineering, Certification and Digital Solutions for Today's Aerospace
• Developing systems and component level requirements for design and verification within the Systems Engineering and Certification operating group and projects. • Working closely with cross-functional teams to develop qualification plans and artifacts required for compliance to satisfy regulatory agencies.
• Performing investigations into detected threats and using customers’ security products to analyze, contain, and remediate threats in their environments • Providing customers with thorough reports of the actions taken to ensure they understand what you did to clean up their environment and protect them from further damage • Identifying effective response strategies to further enhance Red Canary’s customers’ security posture • Actively engaging and collaborating with the Detection Engineering, Threat Hunting, Intel, and Product Management teams to develop new ways of performing timely remediation of identified threats • Lead team projects and mentor peers, while providing strategic insights, recommendations, and operational support for product improvement and overall team success • Participating in an on-call rotation to provide 24x7 remediation
Lead Forward Deployed Engineer
Symmetry SystemsSecuring Data With Precision and Scale in Modern Enterprises
• Own outcomes for high stakes customer engagements • Lead technical discovery to map a customer’s data and identity reality, including key data stores, flows, access patterns, and risk hotspots. • Drive successful pilots and proofs of concept from blank slate to measurable value, including success criteria, implementation plans, and executive ready outcomes. • Design and implement secure deployments across cloud, on-prem, and hybrid environments, including AWS, Azure, GCP, and on premise patterns. • Integrate Symmetry into customer ecosystems such as identity systems, data platforms, and security tooling, leaning on existing integrations and building new ones when needed. Take advantage of AI tooling (such as Claude code) to expedite integrations. • Translate data security posture and identity risk signals into prioritized remediation plans, including least privilege, dormant access reduction, and anomalous activity response. • Partner with Sales to run technical workshops, threat and risk reviews, and architecture sessions that build trust and momentum. • Handle the hard questions about how this works in real environments and about long term outcome arcs. • Mentor other Forward Deployed Engineers and set the bar for technical excellence, customer empathy, and execution. • Build reusable assets such as deployment playbooks, demo environments, reference architectures, and battle tested troubleshooting guides. • Create tight feedback loops with Product and Engineering, turning patterns from the field into roadmap input and shipped improvements.
Senior Forward Deployed Engineer
ParloaAmplify your customer service experience with an AI agent management platform.
• Own deployment engineering projects: Lead the technical execution (incl. product engineering) of Parloa’s deployments inside large, complex enterprise environments • Design for scale and resilience: Architect deployment solutions that meet enterprise-grade requirements for performance, reliability, and security • Engineer solutions where none exist: Build custom extensions, integrations, and configurations to close product gaps and meet enterprise requirements • Work across systems & stacks: Operate at the intersection of backend engineering, DevOps, and data engineering to ensure seamless delivery • Partner with enterprise teams: Collaborate directly with customer engineering organizations to overcome constraints and deliver measurable outcomes • Guide and mentor others: Collaborate with and provide technical guidance to more junior in-field engineers • Influence at senior levels: Engage with enterprise architects and senior technical stakeholders, discussing solutions and architectures in depth to ensure alignment and credibility • Debug under pressure: Rapidly unblock issues in mission-critical environments and ensure customer success. • Shape the product: Act as the feedback loop between the field and our core engineering teams - turning deployment learnings into platform evolution • Deliver long-term impact: Work closely with Deployment Strategists to ensure scalable architectures and durable customer value




