Pokee AI is looking for an AI Infrastructure Engineer to build and optimize systems that power Pokee's RL-trained AI agents.

Company: Pokee AI

Role: AI Infrastructure Engineer

Location: Remote (US/Singapore Preferred)

Experience:

3+ years of experience in ML infrastructure, ML platform engineering, or a related systems role

Role Focus: • 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 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)

Nice to have:

Experience with on-device / edge inference optimization (GGUF quantization, TensorRT-LLM, CoreML, QNN)
Familiarity with on-premise GPU deployments (NVIDIA DGX, Dell PowerEdge, Lenovo ThinkStation)
Experience supporting RL training loops or online learning systems in production
Background in enterprise software with knowledge of security and compliance frameworks
Contributions to open-source ML infrastructure projects

More skills:

ML serving frameworks (vLLM, TensorRT, Triton, ONNX Runtime), container orchestration (Kubernetes, Docker), cloud infrastructure (AWS, GCP), ML experiment tracking and pipeline orchestration tools (MLflow, Weights & Biases, Airflow)

Other:

Opportunity to join a small, elite team solving one of the hardest problems in AI
Direct impact on the product
Access to cutting-edge research
Opportunity to shape the future of enterprise AI from the ground up

AI Infrastructure Engineer

Skills Required

Description

Experience:

Nice to have:

More skills:

Other:

Prepare for this role

More vLLM jobs