# Machine Learning Engineer

**Company:** [Avride](https://hotfix.jobs/companies/avride)
**Location:** Austin, TX
**Role:** ML Engineering
**Experience:** 3+ years
**Skills:** PyTorch, TensorFlow, JAX, Pyspark, NumPy, Scipy, Python, C++, SQL, TensorRT, Triton, Convolutional Neural Networks, Transformers, Multimodal Large Language Models
**Posted:** 2026-04-10

> Develops, optimizes, and deploys deep learning models for autonomous vehicles and robots, managing large datasets and training pipelines. Requires 3+ years experience with PyTorch/TensorFlow, Python, and expertise in computer vision or LLMs.

## Job Description

## What You'll Do

* Develop and Optimize Machine Learning Models: Design, implement, and refine deep learning models to ensure efficiency, scalability, and robustness. This may include developing models for understanding a self-driving vehicle’s surroundings or predicting the intentions of other road users.
* Curate and Manage Large-Scale Datasets: Oversee data collection, preprocessing, and augmentation to maintain high-quality datasets for training and evaluation.
* Enhance and Maintain Training Pipelines: Develop efficient workflows for training, validation, and testing, incorporating distributed training, hyperparameter tuning, and automated monitoring.
* Improve Model Deployment and Efficiency: Optimize inference performance, model compression, and deployment across various hardware platforms.
* Explore and Apply Cutting-Edge ML Techniques: Stay up to date with advancements in deep learning and experiment with novel approaches to improve model performance.
* Collaborate with Cross-Functional Teams: Work closely with researchers, software engineers, and robotics experts to integrate machine learning solutions into real-world autonomous systems.

## What You'll Need

* Strong understanding of fundamental machine learning algorithms and neural network techniques.
* Expertise in at least one modern machine learning domain, such as computer vision, large language models, or generative AI.
* At least three years of experience developing neural network-based algorithms, including data collection, training, and deployment.
* Proficiency in Python and ML frameworks such as PyTorch, TensorFlow, or JAX, along with PySpark, NumPy, and SciPy.
* Working knowledge of C++ and SQL.
* Ability to quickly absorb new concepts by reviewing research papers, technical reports, and documentation.
* Strong collaboration and communication skills, with the ability to align technical work with business objectives and drive results.

## Nice to Have

* Advanced degree in Computer Science, Machine Learning, Robotics, or a related field.
* Experience developing ML algorithms for autonomous vehicles or robotics applications.
* Familiarity with neural network deployment and optimization tools such as **triton**, **TensorRT**, or similar frameworks.
* Proven ability to set and achieve mid- and long-term goals, prioritize tasks, and meet deadlines independently.
* Experience working in cross-functional teams within a multidisciplinary environment.
* Publications in top-tier ML conferences or contributions to patent applications or ML-related open-source projects.

## Similar roles

- [CoDesign & NextGen Performance Engineer](https://hotfix.jobs/jobs/2cb728b7-a9ea-44b0-b16a-47364021f7be) - Cerebras Systems - Sunnyvale, CA
- [Research Engineer, Privacy](https://hotfix.jobs/jobs/3220c5d0-0e5f-4869-b133-1ed126b4208a) - OpenAI - San Francisco, CA - $380k – $445k/yr
- [Research Engineer](https://hotfix.jobs/jobs/b4d18018-e64d-4828-a8b4-303225328e17) - Console - San Francisco, CA - $200k – $350k/yr
- [Software Engineer, AI Platform](https://hotfix.jobs/jobs/7ddad000-3e1c-47cb-a45b-31d8e72abc4f) - Notion - San Francisco, CA - $180k – $201k/yr
- [Machine Learning Engineer](https://hotfix.jobs/jobs/97713038-58cb-46c5-a9c7-0e1554e1ba4f) - Liftoff - Remote - $215k – $275k/yr

**Apply:** https://hotfix.jobs/jobs/fe2a3a0b-9998-4468-abc9-9ec9c06d250b
**Canonical:** https://hotfix.jobs/jobs/fe2a3a0b-9998-4468-abc9-9ec9c06d250b