- 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
- Meta (Redmond, WA)
- …reconstruction, and neural rendering 13. Strong programming experience using python and pytorch 14. Demonstrated software engineer experience via an internship, ... Reinforcement Learning 8. Experience working within Python environments such as pytorch 9. Experience working in a Unix environment **Preferred Qualifications:**… more
- Meta (Redmond, WA)
- …and tracking, generative modeling 11. Strong programming experience using python and pytorch 12. Demonstrated software engineer experience via an internship, ... and design to evaluation 6. Experience working within Python environments such as pytorch 7. Experience working in a Unix environment 8. Must obtain work… more
- Meta (Redmond, WA)
- …learning, generative modeling etc 14. Strong programming experience using python and pytorch 15. Demonstrated software engineer experience via an internship, ... training, benchmark and evaluation 7. Experience working within Python environments such as pytorch 8. Experience working in a Unix environment 9. Must obtain work… 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
- Meta (Menlo Park, CA)
- **Summary:** This role is about developing the core PyTorch 2.0 technologies, innovating and advancing the state-of-the-art of ML compilers, and accelerating PT2 ... adoption through direct engagements with OSS and industry users.The PyTorch Compiler team is dedicated to making PyTorch run faster and more resource-efficient… 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
- Meta (Charleston, WV)
- …& Inference Accelerator) Software team and part of the bigger industry-leading PyTorch AI framework organization. MTIA Software Team has been developing a ... Meta.You will be working on one of the core areas such as PyTorch framework components, AI compiler and runtime, high-performance kernels and tooling to accelerate… more
- Meta (Menlo Park, CA)
- …multi-node data communication through HPC-style collectives. NCCL has been integrated into PyTorch and is on the critical path of multi-GPU distributed training. In ... features, SW benchmarks, performance tuners and SW stacks around NCCL and PyTorch to improve the full-stack distributed ML reliability and performance (eg… 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