AI Infra工程师

【岗位职责】 1.负责根据公司现有数据格式和存储格式，设计并优化最适合训练任务的数据加载方案，包括但不限于 HDF5、JPEG、Parquet、WebDataset、Numpy、S3 / JuiceFS / Ceph 等格式与存储系统。 2.负责优化 PyTorch DataLoader、数据预取、缓存、shuffle、batching、num_workers、pin_memory、异步加载、CPU-GPU 数据流水线等训练数据加载链路，降低 I/O stall，提高 GPU 利用率。 3.负责定位训练过程中的性能瓶颈，包括 GPU 利用率低、通信瓶颈、存储 I/O 瓶颈、CPU 解码瓶颈、Checkpoint 慢、数据加载慢、多机多卡效率低等问题，并给出可落地的优化方案。 4.负责基于公司业务形态，沉淀标准化训练框架和训练模板，支持 VLA / VLM / LLM / Policy / …… 完整岗位描述与官方投递入口,请登录后在岗位详情页查看。

本站当前在招 3096 个上海技术研发岗位，其中 104 个公开标注了薪资。按这些标注统计：月薪（税前）主流区间 12K ~ 22K（25~75 分位），中位约 20K，最低 4K、最高 100K。