- 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 ... Collaborate with the deep learning community to integrate TensorRT to PyTorch . Identify performance opportunities and optimize SoTA models across the spectrum… more
- NVIDIA (Santa Clara, CA)
- We are now looking for a Senior Deep Learning Software Engineer , PyTorch . NVIDIA is hiring software engineers to design and build tools used by AI engineers ... NVIDIA's AI platform as well as directly contributes to PyTorch , a premiere deep learning framework. In this role...Autonomous Vehicles. If you're a creative and motivated software engineer with a real passion for building fast software… more
- Amazon (Seattle, WA)
- …scale. Technical Impact You'll Drive: * Spearhead distributed inference architecture for PyTorch and JAX using XLA * Engineer breakthrough performance ... AWS's next-generation AI accelerators Inferentia and Trainium. As a Senior Software Engineer in our Machine Learning Applications team, you'll be at the forefront… more
- Amazon (Cupertino, CA)
- …degree in computer science or equivalent - Preferred previous software engineer expertise with Pytorch /Jax/Tensorflow, Distributed libraries and Frameworks, ... Web Services (AWS) is looking for a Software Development Engineer II to build, deliver, and maintain complex products...help lead the efforts building distributed training support into Pytorch , Tensorflow using XLA and the Neuron compiler and… more
- NVIDIA (Santa Clara, CA)
- …experience in developing and deploying scalable LLM models, and enterprise applications using Pytorch or TensorFlow with Software Engineer or ML Engineer ... background. + Proven track record of building enterprise RAG-based systems using open-source frameworks such as LlamaIndex, LangChain, Malevis, Haystack etc. + Excellent practical knowledge of Generative AI and LLMs. Ability to train and fine-tune GPT based… more
- LinkedIn (Mountain View, CA)
- …performance optimizations across billions of user queries Model Training Infrastructure: As an engineer on the AI Training Infra team, you will play a crucial role ... resolve issues in popular libraries like Huggingface, Horovod and PyTorch , enable distributed training over 100s of billions of...the process for model training and serving. As an engineer in the team, you will explore and innovate… more
- LinkedIn (Mountain View, CA)
- …performance optimizations across billions of user queries. Model Training Infrastructure: As an engineer on the AI Training Infra team, you will play a crucial role ... resolve issues in popular libraries like Huggingface, Horovod and PyTorch , enable distributed training over 100s of billions of...the process for model training and serving. As an engineer in the team, you will explore and innovate… more
- LinkedIn (Mountain View, CA)
- …performance optimizations across billions of user queries. Model Training Infrastructure: As an engineer on the AI Training Infra team, you will play a crucial role ... resolve issues in popular libraries like Huggingface, Horovod and PyTorch , enable distributed training over 100s of billions of...complex models across LLM and Personalization models. As an engineer , you will build compute efficient infra on top… more
- Google (Sunnyvale, CA)
- Staff Software Engineer , TorchTPU _corporate_fare_ Google _place_ Sunnyvale, CA, USA **Advanced** Experience owning outcomes and decision making, solving ambiguous ... on and is growing every day. As a software engineer , you will work on a specific project critical...TPU Pods) to run training and inference workloads using PyTorch and JAX. Our team is responsible for the… more
- Google (Sunnyvale, CA)
- Senior Software Engineer , TorchTPU _corporate_fare_ Google _place_ Sunnyvale, CA, USA **Mid** Experience driving progress, solving problems, and mentoring more ... TPU Pods) to run training and inference workloads using PyTorch and JAX. In this role, you will be...In this role, you will be responsible for the PyTorch ML framework, processes, ecosystem, and model performance, as… more
- General Motors (Sunnyvale, CA)
- …AI research and model development initiatives. As a Staff ML System Engineer , you will collaborate closely with machine learning engineers, research scientists, and ... + Strong programming skills in Python, with proficiency in frameworks such as, PyTorch (prefered), TensorFlow, or similar + Experience with distributed computing, GPU… more
- LinkedIn (Mountain View, CA)
- …and advanced LLM Agents work for Training infrastructure. As a Principal Staff Software Engineer on the AI Training Infra team, you will play a crucial role in ... identify and resolve issues in popular libraries like Huggingface, Horovod and PyTorch , debug and optimize deep learning training, and provide advanced support for… more
- Amazon (Cupertino, CA)
- …and application framework that seamlessly integrates with popular ML frameworks like PyTorch and JAX enabling unparalleled ML inference and training performance. The ... AWS's custom ML accelerators. Working across the stack from PyTorch till the hardware-software boundary, our engineers build systematic...across the stack from system level optimizations through to Pytorch or JAX is a must have. You can… more
- Amazon (Cupertino, CA)
- …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 ... leading ML frameworks, delivering an outstanding user experience for PyTorch and JAX ML model development on the Trainium...for AWS's accelerator architectures, and engage closely with the PyTorch and JAX and other ML Framework communities to… more
- REI (Seattle, WA)
- **Overview** We are seeking a highly skilled and experienced **Senior Principal Engineer ** to lead our Agentic and Generative AI initiatives. In this role, you will ... both customer and employee experiences. As a **Senior Principal Engineer ** , your responsibilities will include: + **Strategy &...with AI/ML frameworks and libraries such as TensorFlow and PyTorch . + Deep proficiency in Python, with a strong… more
- NVIDIA (Santa Clara, CA)
- …with open-sourced inference frameworks. Seeking a Senior Deep Learning Algorithms Engineer to improve innovative LLMs, VLMs, and multi-modality models. In this ... Optimizer, Megatron-LM, Megatron-Bridge, Nvidia-NeMo, NeMo-AutoModel, TensorRT-LLM) and open-source frameworks ( PyTorch , Hugging Face, vLLM, SGLang). You may also dive… more
- Amazon (Seattle, WA)
- …an ML compiler, runtime and natively integrates into popular ML frameworks, such as PyTorch , TensorFlow and JAX. AWS Neuron is used at scale with customers like ... neural network descriptions created in frameworks such as TensorFlow, PyTorch , and JAX, and converts them into code suitable...quantum leap in performance. As a Machine Learning Compiler Engineer II in the AWS Neuron Compiler team, you… more
- General Motors (Sunnyvale, CA)
- **Job Description** **The Role:** We are seeking a Principal AI Engineer to lead the design and advancement of our AI platform. You will play a key role in shaping ... across distributed training, training efficiency, DDP/FSDP, data processing pipelines, and Pytorch model optimization. This is a highly impactful position where your… more
- Amazon (Seattle, WA)
- …Machine Learning accelerators. This role is for a Senior Machine Learning Engineer in the Distribute Training team for AWS Neuron, responsible for development, ... responsibilities You will lead efforts to build distributed training support into PyTorch , the Neuron compiler, and runtime stacks. You will enable distribute… more
- NVIDIA (Santa Clara, CA)
- …with open-sourced inference frameworks. Seeking a Senior Deep Learning Algorithms Engineer to improve innovative generative AI models like LLMs, VLMs, multimodal ... NVIDIA's ecosystem (TensorRT Model Optimizer, NeMo/Megatron, TensorRT-LLM) and open-source frameworks ( PyTorch , Hugging Face, vLLM, SGLang). You may also dive deeper… more