Skild AI

Software Engineer, AI Training and Infrastructure

Skild AI
Apply
4 months ago
Bengaluru, India +2 moreMid Level
H1B Sponsor

Base Salary

$100k - $300k/yr

Responsibilities

  • Develop and maintain robust, scalable, distributed training pipelines covering data preprocessing, training orchestration, and model evaluation.
  • Optimize training processes for performance, resource utilization, scalability, and reliability.
  • Collaborate with researchers and machine learning engineers to integrate advanced algorithms and techniques into training pipelines.
  • Monitor and analyze training systems, identify bottlenecks, and propose efficiency improvements.
  • Maintain robust and reliable training infrastructure, including automated testing and continuous integration.
  • Explore efficient ways to use diverse data types in the training pipeline.

Requirements

  • A BS, MS, or higher degree in Computer Science, Robotics, Engineering, or a related field, or equivalent practical experience.
  • At least 3 years of industry experience.
  • Proficiency in Python, C++, or a similar language, plus at least one deep learning library such as PyTorch, TensorFlow, or JAX.
  • Strong background in distributed computing, parallel processing, large-scale datasets, and data preprocessing.
  • Deep understanding of current machine learning techniques and models.
  • Experience with cloud-based training environments such as AWS, Google Cloud, or Azure.
  • Experience developing and maintaining machine learning software tooling and infrastructure.
  • Strong understanding and practical experience with algorithms, data structures, system design, and software engineering principles.
  • Experience with continuous integration and automated testing frameworks.

Tech Stack

AWSAzureC++Google CloudPythonPyTorchTensorFlow
Skild AI

About Skild AI

11-50 employees

Building general purpose robotic intelligence.