AI Infrastructure Engineer

Pokee AI is building AI agents trained through reinforcement learning for enterprise use. The AI Infrastructure Engineer will build and optimize scalable training pipelines, high-performance inference serving, data pipelines, and production infrastructure across cloud, on-premise, and on-device environments. The role also involves collaborating with research scientists and ensuring infrastructure reliability, security, and compliance.

Responsibilities:

• Design, build, and maintain scalable training and inference infrastructure for RL-based AI agent models
• Optimize model serving for latency, throughput, and cost across cloud (AWS, GCP) and on-premise/on-device environments
• Develop and manage CI/CD pipelines, experiment tracking, and model versioning systems
• Implement efficient data pipelines for training data collection, preprocessing, and reward signal computation
• Collaborate with research scientists to productionize new algorithms and model architectures
• Ensure infrastructure meets enterprise requirements for reliability, security, and compliance (SOC 2, data residency)