Distinguished Engineer, JAX

EngineerEngineerOtherRemoteLeadTeam 10,001+Since 1993H1B SponsorCompany SiteLinkedIn

Location

California

Posted

147 days ago

Salary

$320K - $488.8K / year

Seniority

Lead

Bachelor Degree18 yrs expEnglishPythonPyTorchscikit-learnTensorFlow

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 Categories

Related Job Pages

More Engineer Jobs

Collinear Group logo

Senior Systems and Certification Engineer

Collinear Group

Engineering, Certification and Digital Solutions for Today's Aerospace

Engineer147 days ago
OtherRemoteTeam 51-200H1B No Sponsor

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

Washington
$140K - $165K / year
Job Closed
Canary Red logo

Senior Threat Response Engineer

Canary Red

Creating the Sounds of Scientific Visualization

Engineer147 days ago
OtherRemoteTeam 1-10H1B No Sponsor

• 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

Colorado
$119K - $150K / year
Job Closed
Symmetry Systems logo

Lead Forward Deployed Engineer

Symmetry Systems

Securing Data With Precision and Scale in Modern Enterprises

Engineer147 days ago
OtherRemoteTeam 11-50H1B No Sponsor

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

United States
Parloa logo

Senior Forward Deployed Engineer

Parloa

Amplify your customer service experience with an AI agent management platform.

Engineer147 days ago
Full TimeRemoteTeam 201-500H1B No Sponsor

• 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

Germany
Job Closed