- 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 (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
- 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