- 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
- 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 (Seattle, WA)
- …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
- Amazon (Cupertino, CA)
- …machine learning accelerators and servers that use them. This role is for a software engineer in the Machine Learning Inference Model Enablement team for AWS ... 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. Experience with Deepspeed and other… 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)
- …Learning workloads?! Join our Developer Tools Always-On Profiling (AON) team as a Senior Software Architect, where you'll be pivotal in designing, implementing, ... from complex multi-node, multi-process, multi-GPU, and cluster environments. + Elevate Software Engineering Excellence: Promote high standards in software … 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)
- …how you can make a lasting impact on the world. We are looking for senior engineers who are mindful of performance analysis and optimization to help us squeeze every ... Services. We are working across all layers of the hardware/ software stack, from GPU architecture to Machine Learning Framework,...TTS, RecSys and others) in multiple ML frameworks (PyT, JAX , TF2, DGL and others) + Implement and test… more
- NVIDIA (Santa Clara, CA)
- …solutions that power some of the world's most advanced computing workloads. We are seeking a Software Engineer to join our MARS team at NVIDIA. In this role, you ... manage data, compute, and networking for large-scale AI workloads. + Build software and automation to orchestrate workloads across thousands of GPUs and petabytes… 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
- NVIDIA (Santa Clara, CA)
- …Ways to stand out from the crowd: + Development experience in storage software such as Key-Value, File systems object storage systems, Vector Databases + Internals ... of frameworks like PyTorch, JAX + Exceptional CUDA programming skills. Exceptional C++ programming skills NVIDIA is widely considered to be one of the technology… more
Related Job Searches:
Engineer,
Senior,
Senior Software Engineer,
Software,
Software Engineer,
Software Senior