Software Engineer, AI Training and Infrastructure

Skild AI
Pittsburgh, San Francisco, Bengaluru
Workplace: OnsiteFull timeUSD 100,000 - 300,000 annuallyFunction: Software EngineeringExperience: 3+ yearsEducation: mastersSkills: ["Communication","Collaboration","Problem-solving"]

Join a robotics-focused AI team to build and optimize robust, scalable training pipelines and frameworks for large-scale models. You’ll collaborate with researchers to integrate state-of-the-art algorithms, improve training performance, and ensure reliable training infrastructure from data prep to deployment across distributed environments.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
Skild AI
Skild AI
1 year ago

Software Engineer, AI Training and Infrastructure

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI
Last checked: 22 hours agoStatus: Live

Job Summary

Join a robotics-focused AI team to build and optimize robust, scalable training pipelines and frameworks for large-scale models. You’ll collaborate with researchers to integrate state-of-the-art algorithms, improve training performance, and ensure reliable training infrastructure from data prep to deployment across distributed environments.
Location: Pittsburgh, San Francisco, Bengaluru
Workplace: Onsite
Employment Type: Full time
Job Function: Software Engineering

Key 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.

Pay and Benefits

Salary: USD 100,000 - 300,000 annually

Key Requirements

  • •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.
  • •Experience with cloud-based training environments (AWS, Google Cloud, Azure).
Experience:3+ yearsRoboticsAIML infrastructureDistributed computing
Education:Master's
Skills:CommunicationCollaborationProblem-solving
Languages:English
Tech Stack:PythonC++PyTorchTensorFlowJAXAWSGoogle CloudAzureContinuous integrationCI

Company Brief

Skild AI
Develops AI-driven solutions focused on applying machine learning to automate and enhance business workflows, delivering predictive insights and intelligent automation for enterprise use cases.
Industry: AI & Machine Learning
Website