> Markdown version of [/jobs/ext/4618605-ai-engineer-data-scaling-optimization](https://www.wearedevelopers.com/jobs/ext/4618605-ai-engineer-data-scaling-optimization). Every page supports `.md` or `Accept: text/markdown`. Links point to the HTML versions so they work for humans too. Agent guide: [/agents.md](https://www.wearedevelopers.com/agents.md).

---

# AI Engineer, Data Scaling & Optimization

- **Company:** AVALA LLC
- **Location:** San Francisco, CA, United States
- **Experience:** Experienced
- **Contract:** Permanent contract
- **Skills:** Artificial Intelligence, Artificial Neural Networks, Computer Vision, Big Data, C++ (Programming Language), Computer Clusters, Software Debugging, Distributed Computing Environment, Distributed Systems, Memory Management, Python (Programming Language), Machine Learning, Tensorflow, Pytorch, Machine Learning Operations, Data Pipelines
- **Published:** November 7, 2026
- **Apply:** https://www.juju.com/job/28_25197_560329a818c8085ef5184d3b353c5f7d

## About the Role

* 4+ years experience with ML infrastructure and training systems
 * Deep expertise in PyTorch or TensorFlow internals
 * Strong understanding of distributed computing and GPU optimization
 * Experience with large-scale data processing (petabyte+)
 * Proficiency in Python and C++ for performance-critical code
 * Knowledge of modern ML architectures (transformers, vision models)
 * Experience optimizing memory usage and training throughput

## Description

* Design and implement high-performance neural network training infrastructure
 * Optimize training and evaluation code for maximum efficiency
 * Build data pipelines handling petabytes of autonomous vehicle and robotics data
 * Develop profiling and debugging tools for ML training systems
 * Implement distributed training across large GPU clusters
 * Work with research teams to enable rapid experimentation at scale