Deskripsi Pekerjaan
Apex Vector Systems is seeking an experienced Senior AI Infrastructure Engineer to architect, scale, and optimize our next-generation distributed deep learning platforms. In this role, you will eliminate bottlenecks across compute, network, and storage layers, directly supporting large-scale foundation model training and inference pipelines.
We offer competitive equity, top-tier health coverage, a flexible hybrid policy, and the opportunity to work at the forefront of AI compute optimization.
Tanggung Jawab
- Architect and maintain high-throughput, low-latency GPU/TPU clusters across hybrid cloud environments.
- Optimize large-scale model training and inference pipelines using Ray, PyTorch, and CUDA-level profiling tools.
- Automate infrastructure provisioning and bare-metal orchestration using Kubernetes and Terraform.
- Implement real-time telemetry, monitoring, and automated remediation for multi-node distributed workloads.
- Collaborate with applied research teams to maximize hardware utilization and reduce training turnaround times.
- Enforce security, access controls, and compliance across all production compute fabrics.
Kualifikasi
- 5+ years of software engineering experience focusing on distributed systems and cloud infrastructure.
- 3+ years of production experience managing Kubernetes clusters and GPU orchestration.
- Proficiency in Python, Go, and shell scripting for systems automation and tooling.
- Hands-on experience with distributed training frameworks such as PyTorch FSDP, DeepSpeed, or Megatron-LM.
- Deep understanding of Linux kernel networking, high-speed fabrics (InfiniBand/RoCE), and distributed storage systems.
- Bachelor's degree in Computer Science, Computer Engineering, or equivalent practical experience.