> Markdown version of [/jobs/ext/961971-software-engineer-ai-training-and-infrastructure](https://www.wearedevelopers.com/jobs/ext/961971-software-engineer-ai-training-and-infrastructure). 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). --- # Software Engineer, AI Training and Infrastructure - **Company:** Skild Ai - **Location:** Pittsburgh, PA, United States - **Experience:** Experienced - **Salary:** $100,000.0 - $300,000.0 - **Contract:** Permanent contract - **Skills:** Artificial Intelligence, Amazon Web Services, Automation of Tests, Microsoft Azure, Big Data, C++ (Programming Language), Cloud Computing, Continuous Integration, Data Cleansing, Data Structures, Distributed Systems, Python (Programming Language), Machine Learning, Software Maintenance, Tensorflow, Software Engineering, Software Systems, Google Cloud, Pytorch, Delivery Pipeline, Deep Learning, Model Validation, Parallel Computation, Information Technology - **Published:** June 9, 2026 - **Apply:** https://www.dice.com/job-detail/59c7b795-4c9d-4ac1-afca-99e9ebf2a5fe ## About the Role * BS, MS or higher degree in Computer Science, Robotics, Engineering or a related field, or equivalent practical experience. * Minimum of 3 years of industry experience. * Proficiency in Python, C++, or similar and at least one deep learning library such as PyTorch, TensorFlow, JAX, etc. * Strong background in distributed computing, parallel processing techniques, handling large-scale datasets and data preprocessing. * Deep understanding of state-of-the-art machine learning techniques and models. * Experience with cloud-based training environments (AWS, Google Cloud, Azure). * Experience in developing and maintaining software tooling and infrastructure for machine learning. * Deep understanding and practical experience with software engineering principles, including algorithms, data structures, and system design. * Experience with continuous integration and automated testing frameworks. ## Description We are looking for a Software Engineer to work at the forefront of developing and optimizing the software infrastructure and tools necessary for training cutting-edge AI models. You will focus on building robust, scalable, and efficient training pipelines and frameworks that support the entire machine learning lifecycle, from data preparation to model deployment. You will collaborate with researchers and machine learning engineers to ensure seamless integration and operation of training systems, pushing the boundaries of what AI can achieve in real-world robotics applications. You will explore new ways to efficiently make use of many types of data in our training pipeline. Responsibilities * Develop and maintain robust, scalable, and distributed training pipelines (data preprocessing, training orchestration, and model evaluation) and frameworks for large-scale AI models. * Optimize training processes for performance and resource utilization, ensuring scalability and reliability. * Collaborate with researchers and machine learning engineers to integrate state-of-the-art algorithms and techniques into training pipelines. * Monitor and analyze training, identifying bottlenecks and proposing solutions to improve efficiency and performance. * Ensure the robustness and reliability of the training infrastructure, including automated testing and continuous integration. ## Related Videos - [Alibaba Big Data and Machine Learning Technology](https://www.wearedevelopers.com/videos/37-alibaba-big-data-and-machine-learning-technology) - [Developing an AI.SDK](https://www.wearedevelopers.com/videos/198-developing-an-ai-sdk) - [Introduction to Azure Machine Learning](https://www.wearedevelopers.com/videos/368-introduction-to-azure-machine-learning) - [Photonic Computing: Programming a New Class of AI Accelerators (incl. Live Coding)](https://www.wearedevelopers.com/videos/100196-photonic-computing-programming-a-new-class-of-ai-accelerators-incl-live-coding) - [Machine learning in the browser with TensorFlowjs](https://www.wearedevelopers.com/videos/155-machine-learning-in-the-browser-with-tensorflowjs) - [Getting Started with Machine Learning](https://www.wearedevelopers.com/videos/260-getting-started-with-machine-learning) ## Related Articles - [How to Become an AI Engineer](https://www.wearedevelopers.com/magazine/331-how-to-become-an-ai-engineer) - [Why Upskilling And Reskilling is Important For Developers](https://www.wearedevelopers.com/magazine/428-why-upskilling-and-reskilling-is-important-for-developers) - [Navigating the AI Shift](https://www.wearedevelopers.com/magazine/629-navigating-the-ai-shift) - [Got AI ideas but no money? Here are 10 free ways to level up your AI skills with Google Cloud](https://www.wearedevelopers.com/magazine/600-got-ai-ideas-but-no-money-here-are-10-free-ways-to-level-up-your-ai-skills-with-google-cloud) - [What Industries Outside of AI Are Hiring The Most AI Experts?](https://www.wearedevelopers.com/magazine/98-what-industries-outside-of-ai-are-hiring-the-most-ai-experts) - [Highest Paying Tech Companies for Developers](https://www.wearedevelopers.com/magazine/220-highest-paying-tech-companies-for-developers)