- 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 Software Engineer for AI Resiliency. At NVIDIA, we are pushing the boundaries of what's possible in AI. We are currently ... seeking a Senior Software Engineer to lead...integrate resiliency features into AI frameworks like PyTorch and JAX /XLA. + Testing & Automation: Develop and implement tests… more
- Google (Sunnyvale, CA)
- Senior Software Engineer , Machine Learning,...Python, and modern deep learning toolkits like PyTorch or JAX . + 3 years of experience in software ... goes on and is growing every day. As a software engineer , you will work on a...deep understanding of deep learning frameworks (like PyTorch or JAX ), strong coding skills, excellent communication abilities, and a… more
- Google (Sunnyvale, CA)
- Senior Staff Software Engineer , Torch TPU _corporate_fare_ Google _place_ Sunnyvale, CA, USA **Advanced** Experience owning outcomes and decision making, ... interconnected hosts. + Experience in contributing to PyTorch or JAX . **About the job** Google's software engineers...goes on and is growing every day. As a software engineer , you will work on a… more
- NVIDIA (Santa Clara, CA)
- NVIDIA is looking for an expert software engineer to help us expand our catalog of Device eXtension (Dx) APIs for our math libraries. For over a decade, NVIDIA's ... libraries that make all of this possible? The hardware and software accelerated computing ecosystem is constantly evolving, including shifts towards hybrid… more
- Amazon (Cupertino, CA)
- …The Annapurna Labs team at Amazon Web Services (AWS) builds AWS Neuron, the software development kit used to accelerate deep learning and GenAI workloads on Amazon's ... that seamlessly integrates with popular ML frameworks like PyTorch and JAX enabling unparalleled ML inference and training performance. The Inference Enablement… more
- Amazon (Cupertino, CA)
- …The Annapurna Labs team at Amazon Web Services (AWS) builds AWS Neuron, the software development kit used to accelerate deep learning and GenAI workloads on Amazon's ... that seamlessly integrates with popular ML frameworks like PyTorch and JAX enabling unparalleled ML inference and training performance. The Inference Enablement… more
- NVIDIA (Santa Clara, CA)
- …can make a lasting impact on the world. We are looking for outstanding Senior Deep Learning Software Engineers to develop and productize NVIDIA's deep learning ... our partners face during product development and coordinate with other architecture and software teams to develop the best solution for partners working on our… more
- NVIDIA (Santa Clara, CA)
- We are now looking for a Senior Deep Learning Software Engineer , PyTorch-TensorRT Performance! NVIDIA is seeking an experienced Deep Learning Engineer ... development for Deep Learning Inference and is seeking excellent Software Engineers at all levels of expertise to join...skills + Experience with a DL framework like PyTorch, JAX , TensorFlow. + Experience with performance analysis and performance… more
- NVIDIA (Santa Clara, CA)
- We are now looking for a Senior Deep Learning Software Engineer , LLM Performance! NVIDIA is seeking an experienced Deep Learning Engineer passionate ... development for Deep Learning Inference and is seeking excellent Software Engineers at all levels of expertise to join...skills + Experience with a DL framework like PyTorch, JAX , TensorFlow. Ways to stand out from the crowd:… more
- NVIDIA (Santa Clara, CA)
- …can make a lasting impact on the world. We are looking for outstanding Senior Deep Learning Software Engineers to develop and productize NVIDIA's deep learning ... our partners face during product development and coordinate with other architecture and software teams to develop the best solution for partners working on our… 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
- Google (Sunnyvale, CA)
- Senior Software Engineer , TorchTPU...to run training and inference workloads using PyTorch and JAX . In this role, you will be responsible for ... debugging, fine tuning). + 5 years of experience in software development. + 5 years of experience testing, and...development. + 5 years of experience testing, and launching software products, and 3 years of experience with … more
- NVIDIA (Santa Clara, CA)
- We are now looking for a Senior Deep Learning Software Engineer , FlashInfer. NVIDIA has been transforming computer graphics, PC gaming, and accelerated ... AI systems engineers to develop groundbreaking technologies in the inference systems software stack! We build innovative AI systems software to accelerate… more
- NVIDIA (Santa Clara, CA)
- …and see how we can make a lasting impact on the world. We are seeking a software engineer to join the Isaac Lab team and propel our flagship platform for robot ... in software development with Python and the deep-learning software stack (Pytorch, Tensorflow, Jax , etc.). + 5+ years of experience in robotics and… more
- NVIDIA (Santa Clara, CA)
- We are seeking a Senior Software Engineer to join a new team building the foundational infrastructure for Robotics Research. This new team will work very ... large-scale MLOps and AI infrastructure + Experience with ML frameworks like PyTorch, JAX , or TensorFlow. + Deep understanding of Kubernetes, experience with Ray +… more
- NVIDIA (Santa Clara, CA)
- NVIDIA is now looking for AI Software Engineers for our GenAI Frameworks (Megatron Core (https://github.com/NVIDIA/Megatron-LM/tree/main/megatron/core) and NeMo ... the intersection of AI applications, libraries, frameworks, and the entire software stack. + Innovate and improve model architectures, distributed training… 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
- Amazon (Cupertino, CA)
- …Inferentia (Inf1/Inf2) 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