- 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 looking for a Senior Deep Learning Software Engineer to design and build our automated inference and deployment solution. As part of the team, you ... encompassing developing features in high-level frameworks like PyTorch and JAX , designing and implementing a high-performance execution environment, low-level GPU… 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
- NVIDIA (Santa Clara, CA)
- We are now looking for a Senior Deep Learning Software Development Engineer ! Academic and commercial groups around the world are using GPUs to power a ... with the world's best talent. Join us in building software which will be used by the entire world...shared with the community. As an experienced and skilled engineer , your collaboration in developing new technology will have… more
- NVIDIA (Santa Clara, CA)
- …and engineering teams alike developing best-in-class AI models. We are now looking for a Senior Deep Learning Software Engineer to develop and scale up our ... improve model efficiency as well as developing an innovative software platform (TRT Model Optimizer). Our software ...stand out from the crowd: + Contributions to PyTorch, JAX , or other Machine Learning Frameworks. + Knowledge of… 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 (Sunnyvale, 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 ... in GPU integrating low-level CUDA work into higher-level frameworks (eg, TF, JAX , PyTorch). Google's software engineers develop the next-generation technologies… 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)
- We are looking for an experienced software professional to contribute to design and development of accelerated and distributed implementations of Python APIs for ... libraries (eg NumPy, SciPy), including accelerated implementations (eg CuPy, Jax .NumPy, NumS, cuNumeric) + Experience developing and publishing Python libraries,… 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)
- …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
- 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)
- …in practice. NVIDIA's Deep Learning Architecture and Libraries Group is looking for a software engineer to help us push the boundaries of our performance ... deep learning performance engineers to devise innovative approaches to hardware and software profiling and incorporate them into our internal tools. Your work will… more