HireFT
Browse JobsHow it worksPricingAboutSuccess Stories
    Back to jobs
    AX

    AXQ Capital

    Financial Services

    Head of Machine Learning Infrastructure

    Beijing, ChinaOn-SiteFull-timePosted 2mo ago
    All AXQ Capital jobs

    Job description

    关于我们

    安贤投资(AXQ Capital)是一家全球量化投资机构。我们立足不同地域、资产类别和交易周期,持续构建多元化的投资优势。我们的策略植根于严谨的量化研究与深入的市场理解,并通过对数据、技术和人工智能的长期投入不断强化。我们汇聚来自全球顶尖高校及领先量化机构的优秀人才,倡导以好奇心、严谨精神和主人翁意识为基础的合作文化。在这里,最好的想法能够脱颖而出,每个人也能与公司共同成长。

    岗位使命

    作为 Head of Machine Learning Infrastructure,你将向 CTO 汇报,负责公司面向机器学习和量化研究的核心计算与平台基础设施,包括 GPU/CPU 集群、分布式训练、研究计算平台、机器学习平台及开发者工具。
    核心目标是打造一套能够支撑大规模特征工程与机器学习训练的基础设施体系,支持 PB 级数据处理能力,并能够高效利用数十 TB 级别训练数据进行模型训练与实验迭代。系统需具备高性能、可扩展、可复现、易使用且成本高效的特性,持续提升研究与工程效率。

    主要职责

    • 负责机器学习基础设施团队建设、技术规划和项目推进,制定中长期平台路线。
    • 建设和优化 GPU/CPU 集群、任务调度和分布式计算平台,支持大规模模型训练、特征计算和量化研究。
    • 构建支持 PB 级数据规模的特征工程与数据处理平台,提升海量数据的计算、存储与访问效率。
    • 支持基于数十 TB 级训练数据的分布式训练与实验体系,优化数据加载、并行训练与资源调度效率。
    • 建设统一的机器学习平台,包括开发环境、实验管理、模型版本、依赖管理和模型部署流程。
    • 优化训练与研究场景下的存储、网络、数据管道和计算性能,提高 GPU/CPU 资源利用率并控制基础设施成本。
    • 建设模型训练、批量/在线推理及相关生产基础设施,提升模型从研究到生产的效率和可靠性。
    • 持续优化开发者工具和研究工作流,提升大规模数据与模型实验的迭代效率。
    • 建设平台监控、容量规划、权限、安全和自动化体系,保障大规模系统的稳定性与可维护性。
    岗位要求
    • 8 年以上机器学习基础设施、分布式系统、平台工程、高性能计算或云基础设施相关经验。
    • 有基础设施或平台团队管理经验,以及复杂系统架构和落地能力。
    • 扎实的 Linux、网络、存储、分布式系统和容器技术基础。
    • 熟悉 GPU 计算、分布式训练和大规模机器学习工作负载,对任务调度、数据 I/O、特征工程和系统性能优化有实践经验。
    • 熟悉 Python、Go、C++、Rust、Java 等至少一门编程语言。
    • 熟悉 Kubernetes、Docker、CI/CD、Infrastructure as Code、监控和云计算基础设施。
    • 具备较强的平台产品意识,能够深入理解量化研究员、机器学习研究员和工程师在大规模数据与模型训练场景下的需求。
    加分项
    • 有量化投资、对冲基金、机器学习平台或大规模研究计算平台经验。
    • 有 GPU 集群、分布式训练或 HPC 平台建设经验。
    • 熟悉 PyTorch Distributed、Ray、Slurm、NCCL、CUDA、RDMA 等技术。
    • 有大规模特征工程平台、数据管道系统或 PB 级数据处理经验。
    • 有 AWS、GCP、阿里云、腾讯云等云平台的 GPU 资源和成本优化经验。
    • 熟悉 Claude Code、Codex、Cursor 等 AI coding tools,并有推动其团队化应用的经验。

    Job details are sourced from the employer's original posting.

    Open job posting
    AX

    About the company

    AXQ Capital

    AXQ Capital is a quantitative trading firm that leverages advanced technology and data science to achieve superior investment returns.

    View all AXQ Capital jobs
    Industry
    Financial Services
    Open roles
    20

    Interested in this role?

    Apply with HireFT

    Free to start — no card required.

    Your fit

    How well do you match?

    Sign in to see how your résumé lines up with this role.