JAX Jobs

41 - 60 of 194 Jobs

Principal Software Engineer, TensorRT-LLMJob Title - Principal Software Engineer, TensorRT-LLM

NVIDIA CorporationCompany Name - NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

We are now looking for a Principal Software Engineer, TensorRT-LLM ! NVIDIA is hiring experienced principal software engineer for its TensorRT-LLM team. Academic and commercial groups around the world are using GPUs to power a revolution in AI, enabling breakthroughs in areas like content creation, code generation and information synthesis, that have put AI on the precipice of an "iPhone moment". Join the team building the AI serving software which is foundational to product lines within NVIDIA

Senior GenAI Algorithms EngineerJob Title - Senior GenAI Algorithms Engineer

NVIDIA CorporationCompany Name - NVIDIA Corporation

•

Remote or Santa Clara, California, USA

Full-time

We are now looking for a Senior Gen AI Algorithms Engineer! NVIDIA is seeking engineers to design, develop and optimize Artificial Intelligence solutions to diverse real-world problems. If you have a strong understanding of deep learning and in particular large language models and their multimodal variants, then this role may be a great fit for you! Collaborate and interact with internal partners, users, and members of the open source community to analyze, define and implement highly optimized A

Senior Deep Learning Software EngineerJob Title - Senior Deep Learning Software Engineer

NVIDIA CorporationCompany Name - NVIDIA Corporation

•

Santa Clara, California, USA

Full-time

We are looking for a Senior Deep Learning Software Engineer to design and build our automated inference and deployment solution. As part of the team, you will be instrumental in defining a scalable architecture for DL inference with emphasis on ease-of-use and compute efficiency. Your work will span multiple layers of the DL deployment stack, encompassing developing features in high-level frameworks like PyTorch and JAX, designing and implementing a high-performance execution environment, low-le

AIML - Staff Machine Learning Engineer, Siri and Information IntelligenceJob Title - AIML - Staff Machine Learning Engineer, Siri and Information Intelligence

Apple, Inc.Company Name - Apple, Inc.

•

No location provided

Full-time

The AIML Information Intelligence team is creating groundbreaking technology for artificial intelligence, machine learning and natural language processing! The features we create are redefining how hundreds of millions of people use their computers and mobile devices to search and find what they are looking for. Our universal search engine powers search features across a variety of Apple products, including Siri, Spotlight, Safari, Messages and Lookup. We also develop state-of-the-art generative

Software Engineer, Machine Learning, CalibrationJob Title - Software Engineer, Machine Learning, Calibration

Nuro Inc.Company Name - Nuro Inc.

•

Mountain View, California, USA

Full-time

Who We Are Nuro exists to better everyday life through robotics. Founded in 2016, Nuro has spent eight years developing autonomous driving (AD) technology and commercializing AD applications. The Nuro Driver is our world-class autonomous driving system that combines AD hardware with our generalized AI-first self-driving software. Built to learn and improve through data, the Nuro Driver is one of the few driverless autonomous technologies on public roads today. Nuro has raised over $2B in capi

AIML - Machine Intelligence - Machine Learning ResearcherJob Title - AIML - Machine Intelligence - Machine Learning Researcher

Apple, Inc.Company Name - Apple, Inc.

•

No location provided

Full-time

The Machine Intelligence, Neural Design (MIND) team is leading Apple-wide innovation on HW/SW co-design for efficient inference. We are looking for a highly motivated, result-oriented team player with strong background in ML/systems and hand-on experience in software development to join our team. Our mission is to advance the state-of-art in ML optimization techniques for efficient inference. Thus, the ideal candidate for this position is a researcher and engineer who will relentlessly push the

Sr. ML Infrastructure Engineer, Apple Data PlatformJob Title - Sr. ML Infrastructure Engineer, Apple Data Platform

Apple, Inc.Company Name - Apple, Inc.

•

No location provided

Full-time

The Apple Data Platform (ADP) group builds the data platform that enables the next generation of intelligent experiences on all Apple products and services. ADP empowers Apple engineers to deliver ML-driven products and innovations rapidly and at scale. We are looking for an experienced engineer who can bring their passion for machine learning, infrastructure, big data, and distributed systems to build world class data+ML platform/products at scale. You will work with many cross functional teams

Model Optimization Engineer (Quality and ML DevOps)Job Title - Model Optimization Engineer (Quality and ML DevOps)

Apple, Inc.Company Name - Apple, Inc.

•

No location provided

Full-time

Are you excited about the impact that optimizing deep learning models can have on enabling transformative user experiences? The field of ML compression research continues to grow rapidly and new techniques to perform quantization, pruning etc are increasingly available to be ported and adopted by the ML developer community, that is looking to ship more models in a constrained memory budget and make them run faster. We are passionate about productizing and pushing the envelope of the state of the

AIML - Machine Learning Engineer, Foundation ModelsJob Title - AIML - Machine Learning Engineer, Foundation Models

Apple, Inc.Company Name - Apple, Inc.

•

No location provided

Full-time

We are a group of engineers and researchers responsible for building foundation models at Apple. We build infrastructure, datasets, and models with fundamental general capabilities such as understanding and generation of text, images, speech, videos, and other modalities and apply these models to Apple products. We are looking for engineers who are passionate about building systems that push the frontier of deep learning in terms of scaling, efficiency, and flexibility and delight millions of us

Senior Software Engineer, Machine Learning, Sensor SimulationJob Title - Senior Software Engineer, Machine Learning, Sensor Simulation

Nuro Inc.Company Name - Nuro Inc.

•

Mountain View, California, USA

Full-time

Senior ML Software EngineerJob Title - Senior ML Software Engineer

NVIDIA CorporationCompany Name - NVIDIA Corporation

•

Santa Clara, California, USA

Full-time

At NVIDIA, we're building the foundation for the next generation of Physical AI. Our Cosmos team is passionate about developing robust, production-grade open-source ML software to support machine learning and gen-AI research at scale. We're looking for expert-level Python engineers who are passionate about building production-ready systems and want to make a lasting impact through open-source contributions. If you care deeply about software craftsmanship, maintainability, and performance-and ha

Senior Deep Learning Compiler Engineer - CUDAJob Title - Senior Deep Learning Compiler Engineer - CUDA

NVIDIA CorporationCompany Name - NVIDIA Corporation

•

Seattle, Washington, USA

Full-time

NVIDIA has been transforming computer graphics, PC gaming, and accelerated computing for more than 25 years. It's a unique legacy of innovation that's fueled by great technology-and amazing people. Today, we're tapping into the unlimited potential of AI to define the next era of computing. An era in which our GPU acts as the brains of computers, robots, and self-driving cars that can understand the world. Doing what's never been done before takes vision, innovation, and the world's best talent.

Electromechanical Technician (JAX2 - Jacksonville, FL)Job Title - Electromechanical Technician (JAX2 - Jacksonville, FL)

Apex SystemsCompany Name - Apex Systems

•

Jacksonville, Florida, USA

Full-time

Job#: 2079880 Job Description: Apex Systems is seeking multiple Electro-Mechanical Technicians to join our client located in Jacksonville, FL. We intend to interview prospective candidates as early as this week. This is a 2-3 month contract opportunity. Compensation: $25/hr (overtime rate of "time and a half") Schedule: M-F 10 hours shifts for a total of 50 hours per week expected Benefits: Access to health, dental, vision, 401k, and a weekly pay structure For immediate consideration, please se

Machine Learning Kernel Performance Engineer, DojoJob Title - Machine Learning Kernel Performance Engineer, Dojo

Tesla MotorsCompany Name - Tesla Motors

•

Palo Alto, California, USA

Full-time

As a member of the Dojo Machine Learning software team, you will be responsible for developing and optimizing machine learning kernels to run on a massively parallel machine. The ideal candidate will have a strong background in kernel development and performance optimization for AI workloads, with a passion for delivering high-performance implementations, which are simple to use. Responsibilities Develop and validate datapath kernels on a massively parallel machineDesign and deliver implementat

Senior Python Machine Learning EngineerJob Title - Senior Python Machine Learning Engineer

EPAM SystemsCompany Name - EPAM Systems

•

No location provided

Full-time

We seek a highly skilled Python Machine Learning Engineer with extensive experience designing, developing and deploying Machine Learning models and NLP applications. As a Python Machine Learning Engineer, you will provide expertise in LLMs for diverse Python applications, such as chatbots, question-answering systems, and content generation. Immerse yourself in our collaborative culture by working on-site at our modern offices in Tbilisi and Batumi or a coworking space in Kutaisi. #LI-DNI#top-va

AIML - Machine Learning Engineer, Siri and Information IntelligenceJob Title - AIML - Machine Learning Engineer, Siri and Information Intelligence

Apple, Inc.Company Name - Apple, Inc.

•

No location provided

Full-time

Do you want to make Siri and Apple products smarter for our users? The Siri and Information Intelligence team is redefining how hundreds of millions of people use their devices to get information. We are an Applied ML team pushing the limits of question answering, assistant response ranking, summarization, and search technologies, while also responsible for a production service. We are part of a wider effort to power information across a variety of Apple products - including Siri, Spotlight, Saf

Senior DGX AI Cloud Performance Analysis Tools EngineerJob Title - Senior DGX AI Cloud Performance Analysis Tools Engineer

NVIDIA CorporationCompany Name - NVIDIA Corporation

•

Santa Clara, California, USA

Full-time

Joining NVIDIA's DGX Cloud AI Efficiency Team means contributing to the infrastructure that powers our innovative AI research. This team focuses on optimizing efficiency and resiliency of AI workloads, as well as developing scalable AI and Data infrastructure tools and services. Our objective is to deliver a stable, scalable environment for AI researchers, providing them with the necessary resources and scale to foster innovation. We are seeking excellent Software Engineers to design and develop

On-device ML Infrastructure Engineer (ML User Experience APIs)Job Title - On-device ML Infrastructure Engineer (ML User Experience APIs)

Apple, Inc.Company Name - Apple, Inc.

•

No location provided

Full-time

The On-Device Machine Learning team at Apple is responsible for enabling the Research to Production lifecycle of cutting edge machine learning models that power magical user experiences on Apple's hardware and software platforms. Apple is the best place to do on-device machine learning, and this team sits at the heart of that discipline, interfacing with research, SW engineering, HW engineering, and products. The team builds critical infrastructure that begins with onboarding the latest machine

Compiler Engineer (Entry Level)Job Title - Compiler Engineer (Entry Level)

Ericsson Inc.Company Name - Ericsson Inc.

•

Austin, Texas, USA

Full-time

Join our Team Sponsorship: Ericsson Inc. does not sponsor US work authorizations for this job position including H-1B, O-1, L-1, and TN. Ericsson also does not hire F-1's working onfor this position. Job Title: Compiler Engineer (Entry Level) Location: Austin Texas (Hybrid) About this opportunity: Ericsson is looking for a passionate and motivated early-career software engineer to join our Systems team, focusing on machine learning compiler development. This entry-level role is ideal for ca

Software Developer, AI NetworkingJob Title - Software Developer, AI Networking

NVIDIA CorporationCompany Name - NVIDIA Corporation

•

Santa Clara, California, USA

Full-time

NVIDIA is changing the world of AI Networking with groundbreaking technology. We are excited to be adding an AI Networking Software Developer to our AI Networking SW development and codesign team. In this role, you will be designing and implementing new protocols and algorithms that enable the best performance possible on NVIDIA networked supercomputers and data. centers. You will be delivering production grade solutions, getting hands on with creative and inventive technology, and collaborating