当前位置: 首页 > 

Kakao Brain公开了AI开发库“Trident”

|
韩国Kakao公司的人工智能(AI)研发子公司“Kakao Brain”8日表示,推出了提高人工智能模型训练和推理速度的“Trident”性能库。开发人员可以在开源平台GitHub上使用该库。

Trident基于Triton实现,Triton是OpenAI的图形处理单元(GPU)编程语言。该库由可重用的代码包组成。Trident通过优化的内核优化了AI模型开发所需的计算。它的用法与PyTorch相同,PyTorch是由meta开发的机器学习框架。

根据该公司的解释,与现有方法相比,使用Trident的人工智能开发人员可以将人工智能模型训练和推理的计算时间减少约25%。Kakao Brain在GitHub上发布了这个库,允许开发人员在他们的人工智能模型研究中使用它。

Kakao Brain代表金一斗表示,“为了支持众多AI开发人员轻松创建大型AI模型,并为大型语言模型和图像生成模型等关键技术的出版等AI生态系统的进步做出持续贡献,决定发布Trident。”

TAG: