Senior/Staff Software Engineer, ML Infrastructure, Optimization
Job in
Mountain View, Santa Clara County, California, 94039, USA
Listed on 2026-08-07
Listing for:
Towards AI, Inc.
Full Time
position Listed on 2026-08-07
Job specializations:
-
Software Development
Machine Learning/ ML Engineer, AI Engineer (Applied/Software), Data Scientist
Job Description & How to Apply Below
194K - 352K USD Annual Full Time## Role Highlights### Languages used
PythonC++CUDA### Key skills
Deep Learning Machine Learning Full StackLLMsAILogistics Deployment Infrastructure Optimization Perception Compilers Robotics ### Tools, Libraries and FrameworksGPUsJAXPy Torch Tensorflow Keras ## Description The role involves designing, building, and deploying core infrastructure components within the machine learning model life cycle. Responsibilities include optimizing the autonomy stack through techniques such as quantization, distillation, and model compression. The individual will work with autonomy engineers to validate and deploy large language models.
Furthermore, the position requires developing and maintaining a model compiler framework. The work also entails writing software to ensure vehicle safety and collaborating with domain experts to implement end-to-end machine learning solutions.##
Required Qualifications and Skills Candidates must possess at least three years of relevant experience in machine learning optimization infrastructure. Proficiency in Python and working experience with C++ and CUDA are required. Applicants must demonstrate experience with machine learning optimization techniques, including quantization and pruning, as well as experience with machine learning compilers. Additionally, the role requires working experience with deep learning frameworks such as PyTorch, Jax, Tensorflow, and Keras.
#J-18808-Ljbffr
Position Requirements
10+ Years
work experience
To View & Apply for jobs on this site that accept applications from your location or country, tap the button below to make a Search.
(If this job is in fact in your jurisdiction, then you may be using a Proxy or VPN to access this site, and to progress further, you should change your connectivity to another mobile device or PC).
(If this job is in fact in your jurisdiction, then you may be using a Proxy or VPN to access this site, and to progress further, you should change your connectivity to another mobile device or PC).
Search for further Jobs Here:
×