Senior Software Engineer, JAX

Full-stack EngineerSoftware EngineerFull TimeRemoteSeniorTeam 10,001+Since 1993H1B SponsorCompany SiteLinkedIn

Location

California

Posted

2 days ago

Salary

$184K - $356.5K / year

Seniority

Senior

Bachelor Degree6 yrs expEnglishPythonPyTorchScikit-LearnTensorflow

Job Description

Senior Software Engineer, JAX

NVIDIA

• Play meaningful 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)
  • 6+ 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 Job Pages

More Full-stack Engineer Jobs

Full TimeRemoteTeam 11-50Since 2000H1B Sponsor

• Design, build, and deliver large-scale and real-time systems with high quality, scalability, and operability in mind. • Collaborate with product management, data engineering, and other engineering teams to bring end-to-end solutions to our customers. • Review and provide feedback on technical designs and code changes to engineering peers. • Proactively identify areas of technical debt and propose solutions to address them. • Take operational ownership of the system by using monitoring and observability tools to identify and respond to issues. • Mentor and coach team members to raise the technical bar on the team.

United States
$170K - $220K / year
Full TimeRemoteTeam 51-200Since 2017H1B No Sponsor

• Support consumer-facing systems for the school store and kids club products • Contribute to new school fundraising initiatives • Design and build engaging user experiences to drive customer growth and retention • Collaborate with a small team and industry experts in machine learning and personalization

United States
$140K - $150K / year
Hype Empreendimentos logo

Senior Full-Stack Developer

Hype Empreendimentos

Proporcionamos diariamente Experiências Extraordinárias!

Full TimeRemoteTeam 201-500Since 2009H1B No Sponsor

• Design, build, and maintain full-stack applications (back-end in Node.js/Python and front-end in React/Next.js); • Define solution architecture, including data modeling, integrations, and selection of technical patterns; • Develop and maintain REST and GraphQL APIs for system communication; • Manage data storage in PostgreSQL, MongoDB, and Redis; • Integrate and evolve AI agent features (OpenAI, Anthropic); • Build and maintain infrastructure on AWS (S3, EC2, Lambda) and container platforms (Docker/Kubernetes); • Conduct code reviews, define best practices, and mentor less-experienced developers; • Participate in architecture decisions and technical priorities alongside Tech leadership; • Collaborate with business teams to gather requirements and propose technical solutions.

Brazil
Full TimeRemoteTeam 51-200H1B Sponsor

• Lead OSP due diligence. Internal authority on outside-plant due diligence across new deployments — site assessment, provider qualification, and risk identification. • Distinguish yourself as a design team leader and liaison to Mechanical, Electrical, and Civil disciplines for coordination on connectivity pathways. Help set our team up for success as an inside authority on clear pathway requirements and constraints. High visibility for upward mobility and promotion as our business grows rapidly • Fiber distribution design. Own fiber distribution design and iterate project-to-project based on deployment-specific factors including scale, phasing, and site topology. • Own conduit sizing decisions across fiber ingress, inter-datahall connections, and pathway type selection. Oversee pre-cabled trunk sizing. • Cable tray loading and bonding/grounding coordination. Own cable tray loading calculations. Coordinate between the OSP provider and Armada's electrical engineers overseeing telecom bonding and grounding (TGB, TBB). • Oversee pre-construction submittals, including action submittals (technical approvals, product data, drawings, etc) to ensure compliance with project requirements. • Resume building opportunity to support SE lead technical sales motions for highest impact campus opportunities. Planned deployments of 20MW-50MW and greater capacities, deal sizes on average 200MM and greater

United States
$137.0K - $171.3K / year