- NVIDIA (Santa Clara, CA)
- NVIDIA is hiring senior engineers to develop its AI platform and more specifically its performance optimizations in deep learning frameworks using JAX , a tool ... for a wide range of deep learning solutions using JAX . To achieve this and provide an excellent user...foundation in CPU and GPU architectures, numeric libraries, modular software design. + Highly motivated with excellent verbal and… more
- Amazon (Cupertino, CA)
- Description AWS Neuron is the complete software stack for the AWS Inferentia and Trainium cloud-scale machine learning accelerators. As a part of the Neuron ... Frameworks team you'll develop and enhance support PyTorch and JAX for AWS Neuron, working with the open source ecosystem. You will develop and extend support for… more
- NVIDIA (Santa Clara, CA)
- We are now looking for a Senior Deep Learning Software Engineer , for Algorithmic Model Optimization! Join our team of algorithmic model optimization experts ... large language models (LLM) and diffusion models. As a Senior Deep Learning Software Engineer ,...stand out from the crowd: + Contributions to PyTorch, JAX , or other Machine Learning Frameworks + Knowledge of… more
- LinkedIn (Mountain View, CA)
- …engineers to optimize their models and deliver the best performance possible. As a Senior Software Engineer , you will have first-hand opportunities to ... optimize performance across algorithms, AI frameworks, data infra, compute software , and hardware to harness the power of our...billions of user queries Model Training Infrastructure: As an engineer on the AI Training Infra team, you will… more
- NVIDIA (Santa Clara, CA)
- …vital resources and scale to champion innovation. We are seeking a distributed software engineer to join our team! As a Senior engineer , you'll be ... like-minded engineers. What You Will Be Doing: As a software engineer specializing in backend development, you'll...world experience in DL frameworks, orchestrators like PyTorch, TensorFlow, JAX , and Ray + Experience in developing a framework… more
- Amazon (Cupertino, CA)
- …accelerators and the Trn1 and Inf2 servers that use them. This role is for a software engineer in the Neuron Inference (ML Apps) team for AWS Neuron. This role ... Description AWS Neuron is the complete software stack for the AWS Inferentia and Trainium...and throughput on these large models using PyTorch or JAX is a must. Experience with technologies/tools such as… more
- Google (Mountain View, CA)
- …UI design and mobile; the list goes on and is growing every day. As a software engineer , you will work on a specific project critical to Google's needs with ... equivalent practical experience. + 5 years of experience with software development in one or more programming languages, and...years of experience in Applied Machine Learning (eg, TensorFlow, JAX , CUDA, etc.). + 1 year of experience in… more
- NVIDIA (Santa Clara, CA)
- …vision, artificial intelligence, and deep learning. Our team builds the accelerated software ecosystem that enables visual AI developers to innovate swiftly and ... domain with strong collaboration and passion for efficient, well-crafted software . Looking for a chance to make gaussian splatting,...common deep learning frameworks such as PyTorch, OnnxRT, or JAX + Current with the latest work in Gaussian… more
- NVIDIA (Santa Clara, CA)
- …PyTorch open source community to deliver our customers the best of NVIDIA software . In this position you will learn innovative techniques from NVIDIA's domain ... written in Numpy and PyTorch. Beyond building nvFuser influence and improve the entire software stack all the way from users to the CUDA compiler, to the… more
- NVIDIA (Santa Clara, CA)
- We are now looking for AI Software Engineers for NeMo! NVIDIA NeMo is an open-source, scalable and cloud-native framework built for researchers and developers ... intersection of deep learning applications, libraries, frameworks, and the entire software stack. + Performance tuning and optimizations of deep learning framework… more
- NVIDIA (Santa Clara, CA)
- …vision, artificial intelligence, and deep learning. Our team builds the accelerated software ecosystem that enables visual AI developers to innovate swiftly and ... experience. + Excellent architectural sense and can-do approach to building high-performance software at scale. Up to date on key technologies such as Kubernetes,… more
- NVIDIA (Santa Clara, CA)
- …Strong Python programming skills. Familiarity with containers, numeric libraries, modular software design + Good knowledge of state-of-the-art DNN architectures and ... with experience in developing or using major deep learning frameworks (PyTorch, Tensorflow, JAX etc.) + Experience with solving and using machine learning for real… more
- NVIDIA (Santa Clara, CA)
- NVIDIA is hiring senior software engineers to build and optimize the tools Deep Learning engineers use across the world to design, develop, and deploy AI ... directly contributes to premiere Deep Learning frameworks - PyTorch, JAX and TensorFlow. We work with multiple teams both...advise design of new hardware generations and core platform software components. What we need to see: + BS… more
- Amazon (Cupertino, CA)
- …accelerators and the Trn1 and Inf1 servers that use them. This role is for a senior software engineer in the Machine Learning Applications (ML Apps) team for ... - Bachelor's degree in computer science or equivalent - Preferred previous software engineer expertise with Pytorch/ Jax /Tensorflow, Distributed libraries and… more
- Amazon (Cupertino, CA)
- …accelerators and the Trn1 and Inf1 servers that use them. This role is for a senior software engineer in the Machine Learning Applications (ML Apps) team for ... - Bachelor's degree in computer science or equivalent - Preferred previous software engineer expertise with Pytorch/ Jax /Tensorflow, Distributed libraries and… more
- Amazon (Cupertino, CA)
- …and Trainium (Trn1), our cloud-scale Machine Learning accelerators. This role is for a senior machine learning engineer in the Distribute Training team for AWS ... science or equivalent - Experience in computer architecture - Previous software engineering expertise with Pytorch/ Jax /Tensorflow, Distributed libraries and… more
- Amazon (Cupertino, CA)
- …and the Trn1 and Inf1 servers that use them. This role is for a software engineer in the Machine Learning Applications (ML Apps) team for AWS Neuron. ... Description AWS Neuron is the complete software stack for the AWS Inferentia and Trainium...throughput on these large models using Python, Pytorch or JAX is a must. Deepspeed and other distributed inference… more
- Amazon (Cupertino, CA)
- …and operations experience - Experience in computer architecture - Previous software engineering expertise with Pytorch/ Jax /Tensorflow, Distributed libraries and ... Labs (our organization within AWS UC) designs silicon and software that accelerates innovation. Customers choose us to create...Learning accelerators. This role is for a machine learning engineer in the Distribute Training team for AWS Neuron,… more
- NVIDIA (Santa Clara, CA)
- We are now looking for a Senior High-Performance LLM Training Engineer ! NVIDIA is seeking experienced engineers specializing in performance analysis and ... systems. This position focuses on optimizing NVIDIA's high-performance LLM software stack in frameworks like PyTorch and JAX... software stack in frameworks like PyTorch and JAX for high-performance training on thousands of GPUs, while… more
- NVIDIA (Santa Clara, CA)
- …with the necessary resources and scale to foster innovation. We are seeking a Senior Site Reliability Engineer (SRE) to join our team. You'll be instrumental ... and performance is part of the role. As a Senior SRE at NVIDIA, you will have the opportunity...Good understanding on DL frameworks, orchestrators like PyTorch, TensorFlow, JAX , and Ray + Strong background in software… more