Apple开源Apple Silicon机器学习框架MLX
站长网2023-12-06 16:49:412阅
Apple 开源的 MLX 是一个适用于苹果芯片的机器学习框架。它具有许多功能,包括熟悉的 API、可组合的函数转换、延迟计算、动态图构建和多设备支持等。
MLX 从 NumPy、PyTorch、Jax 和 ArrayFire 等框架中吸取灵感,创建了一个强大且多功能的平台。通过示例仓库,我们可以看到 MLX 的各种能力,包括语言模型训练、文本生成、图像生成和语音识别。MLX 的安装也非常简单,只需使用 pip 命令即可。
代码: https://github.com/ml-explore/mlx
文档: https://ml-explore.github.io/mlx/build/html/index.html
重点功能:
熟悉的 API:MLX 的 Python API 与 NumPy 紧密结合,而功能齐全的 C API 则反映了 Python 版本。此外,mlx.nn 和 mlx.optimizers 等更高级别的包通过遵守 PyTorch 约定来简化模型构建。
可组合函数转换:MLX 引入了可组合函数转换,支持自动微分、向量化和计算图优化。
惰性计算:MLX 中的计算被设计为惰性计算,确保仅在必要时才具体化数组,从而优化计算效率。
动态图构建:MLX采用动态图构建,消除了因函数参数形状变化而引发的缓慢编译。这种方法简化了调试过程。
多设备支持:MLX 允许操作在支持的设备(包括 CPU 和 GPU)上无缝运行,为开发人员提供了灵活性。
统一内存模型:MLX 引入了与其他框架不同的统一内存模型。阵列驻留在共享内存中,允许跨不同设备类型对 MLX 阵列进行操作,而无需移动数据。
0002
评论列表
共(0)条相关推荐
世界顶级人工智能研究人员齐聚非洲就 AI 技术的未来走向展开辩论
上周,来自世界各地的2000多名研究人员和工程师聚集在卢旺达,就人工智能的未来走向展开竞争性的辩论。一方面,他们拟建更强大的系统,如ChatGPT,旨在超越人类智能,提高工人生产率并增加经济增长;另一方面,也有人拟建更小型,更有针对性的人工智能解决方案,以应对全球性的挑战,如气候变化、改善医疗保健和保护生物多样性。站长网2023-05-08 10:28:320000句子互动融到新钱:真是漫长的准备,一扫私域融资寒冬!
句子互动近日完成数百万美元Pre-A轮融资时,或许没想到,这个动作给行业注入了新活力和希望。近1年来,和私域SaaS、运营工具相关的融资几乎进入寒冬,业界罕有消息传出。但自此开始,行业似乎开启了新的融资周期。“我们一直坚信Chatbot会成为未来营销的基础设施和重要的人机交互窗口,为此我们等了7年”。句子互动创始人李佳芮这样提及他们此前的漫长准备。站长网2023-06-15 19:51:420000德国翻译公司DeepL部署NVIDIA DGX SuperPOD 以提升LLM功能
德国的神经机器翻译服务DeepL宣布在其瑞典的一个数据中心部署了NVIDIAAI数据中心基础设施平台DGXSuperPOD,以提升其LLM功能。这个超级计算机由68个NVIDIADGXH100系统组成,性能达到了21.85PFlop/s,在全球性能排名中位居第26位,在欧洲排名第8位。DeepL表示,这是欧洲首次实现如此规模的商业部署。站长网2023-08-03 10:11:500000小米14系列发布 售价3999元起首发第三代骁龙8
近日,小米举办了主题为“跨越时刻”的新品发布会,正式推出了小米14和小米14Pro。这两款新机均搭载了第三代骁龙8移动平台,并搭配全新的小米澎湃OS操作系统。在设计上,小米14提供四种配色选择,包括岩石青、黑色、白色和雪山粉,而小米14Pro除了提供常规配色外,还推出了一款钛金属特别版。站长网2023-10-27 10:34:260000DCI-VTON:基于扩散模型的高质量虚拟试衣技术 保留衣物细节
DCI-VTON是一种结合了样例基础修复和变形模块的扩散模型,用于实现虚拟试穿任务。它能够将一张图像上的衣物转移到另一张图像上,并在更高分辨率下保留衣物细节,生成逼真的虚拟试穿图像。论文地址:https://arxiv.org/abs/2308.06101v1站长网2023-08-17 11:10:330004