Senior Software Engineer, JAX

🔥 1 hour ago

🏄 California – Remote

🏄 California – Remote

💵 $184k - $356.5k / year

⏰ Full Time

🟠 Senior

🧑‍💻 Full-stack Engineer

🦅 H1B Visa Sponsor

This company has sponsored H1B visas in the past.

Apply Now

Find Similar Remote Jobs

Set Up Email Alert For New Jobs

📊 Check your resume score for this job

Improve your chances of getting an interview by checking your resume score before you apply.

Upload Resume

NVIDIA

WebsiteLinkedInAll Job Openings

10,000+ employees

Founded 1993

🤖 Artificial Intelligence

🎮 Gaming

Artificial Intelligence • Gaming • Automotive

NVIDIA is a leading technology company specializing in accelerated computing and artificial intelligence. NVIDIA pioneers advancements in graphical processing units (GPUs), cloud computing, data centers, and virtual reality, with a focus on gaming, automotive, healthcare, and robotics industries. The company's innovations, such as NVIDIA Omniverse, transform traditional digital processes by enabling high-fidelity simulations and rendering tasks. Their applications span various industries, from autonomous vehicles using NVIDIA DRIVE to healthcare solutions with NVIDIA Clara, and AI-driven analytics and workflows.

📋 Description

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

🎯 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