Jobs
Indexed directly from employers. Every age is their own publish date.
Searching titles and descriptions for “Senior Software Engineer, JAX”. A match may be a passing mention rather than the job itself. Titles only.
13 roles
1ySenior Software Engineer, Training Efficiency…Collaborate with researchers and ML engineers to resolve bottlenecks in data pipeline performance. Improve runtime goodput of ML training workload, including optimizing input data…
- 1.1ySenior Deep Learning Compiler Engineer - PyTorchNVIDIAEngineer, Tech SWBerlin, Berlin; Amsterdam, NH; Warsaw, Masovian Voivodeship; Zürich, ZH; Munich, BYhalf of NVIDIA’s roles are under 3 months
…experience building complex, well-tested software systems. Hands-on experience with deep learning frameworks like PyTorch or JAX. You understand how models are built…
1.1yML Research Engineer, London…Research Engineering (Machine Learning), London We are looking for Research Engineers with different levels of experience - Mid through to Senior, Staff, Principal or equivalent…
- 1.1ySenior Solutions Architect, Financial Services Capital Markets
…Science, Electrical/Computer Engineering, Physics, Mathematics, or other Engineering fields (or equivalent experience) 12+ years experience as an ML/Software Engineer with a proven…
1.2ySenior/Staff Engineer, Machine Learning - Online Mapping…About the Team Our robotics team is growing and we are looking for an ML Software Engineer to join our Online Mapping team. We…
1.3ySr. Software Engineer- AI/ML, AWS Neuron Distributed Training - Performance Optimization…This role is for a software engineer in the Distributed Training team for AWS Neuron. This role is responsible for development, enablement and performance…
- 1.3ySenior System Software Architect, HPC and AI Networking
…Join our world-class team of researchers and engineers building next-generation software and hardware systems that power the most demanding AI workloads on…
- 1.3ySenior Inference Technical Product Marketing Manager - Accelerated Computing
We are looking for a Senior Technical Product Marketing Manager. This role will be located in our rapidly growing data center business and pivotal…
- 1.4yMachine Learning Engineer, Simulation Realism
…ML software engineering 5+ years of experience with Python 5+ years of ML software experience with machine learning framework including Pytorch, Tensorflow, Jax/Flax…
- 1.5ySenior/Staff ML Engineer, 3D/4D World Modeling, Simulation
…Our team is a diverse, and collaborative group of machine learning (ML) engineers, software engineers, and ML research engineers. We develop industry-leading simulation…
1.5yAIML - Senior Machine Learning Infrastructure Engineer -ML Compute, ML Platform & Technology…Here, you’ll do more than join something — you’ll add something! As a Senior Engineer on ML Compute Team, you’ll join a…
- 1.6yMachine Learning Engineer, Runtime & Optimization
…We are looking for engineers with ML software or ML systems expertise to help us improve compute performance on both cloud and car. You…
1.7ySenior Member of Technical Staff, Multimodal AICohereModelingSan Francisco; New York; San Francisco; Paris; Toronto; Montrealhalf of Cohere’s roles are under 48 daysRemote…You are an ideal candidate if you: - Possess exceptional software engineering skills, with a proven track record of building robust and scalable systems. - Have…