Project Icon

mars

多库支持的张量计算框架

Mars是一种基于张量的统一框架,支持大规模数据计算,兼容Numpy、Pandas、Scikit-learn等多个库。无论是单机还是集群环境,Mars都能简化数据处理工作。它提供了详细的安装指南、架构概览和多种使用模式,并与TensorFlow、PyTorch等库深度集成,显著提升计算效率。更多安装和使用信息,请参阅文档。

Mars 项目介绍

Mars 是一个基于张量的统一框架,专注于大规模数据计算。它扩展并支持了包括 Numpy、Pandas、Scikit-learn 在内的多个库的横向扩展能力。Mars 能够在数据处理的同时保持这些库熟悉的接口,提供了简单、强大且便于扩展的大数据处理解决方案。

安装说明

Mars 的安装过程非常简单。要安装 Mars,只需在终端中运行以下命令:

pip install pymars

开发者安装

如果您希望为 Mars 项目贡献代码,您可以通过以下步骤安装开发版本:

git clone https://github.com/mars-project/mars.git
cd mars
pip install -e ".[dev]"

详细的安装说明可以参阅 Mars 的官方文档。

架构概览

Mars 的架构设计支持在多核甚至分布式环境下高效运行,从而增强了运算的速度和计算能力。

入门指南

您可以通过以下方式在本地启动一个新的计算会话:

import mars
mars.new_session()

或者,连接到已经初始化的 Mars 集群:

import mars
mars.new_session('http://<web_ip>:<ui_port>')

Mars 张量

Mars 张量提供了类似于 Numpy 的接口。比较下列代码示例,左侧是 Numpy 的实现,右侧是 Mars 张量的实现:

import numpy as np
import mars.tensor as mt

N = 200_000_000
a = np.random.uniform(-1, 1, size=(N, 2))
print((np.linalg.norm(a, axis=1) < 1).sum() * 4 / N)

Mars 充分利用多核处理,即使在笔记本电脑上也能高效运行,并且在分布式设置下表现尤其出色。

Mars 数据帧

Mars 提供了类似于 Pandas 的数据帧接口。下列代码展示了 Pandas 和 Mars 数据帧的对比:

import numpy as np
import pandas as pd
import mars.tensor as mt
import mars.dataframe as md

df = pd.DataFrame(np.random.rand(100_000_000, 4), columns=list('abcd'))
print(df.sum())

df = md.DataFrame(mt.random.rand(100_000_000, 4), columns=list('abcd'))
print(df.sum().execute())

Mars 学习

Mars 学习部分提供了类似于 Scikit-learn 的接口,并且能够与许多机器学习框架集成,例如 TensorFlow、PyTorch、XGBoost、LightGBM 等。以下是与 Scikit-learn 的一个简单对比:

from sklearn.datasets import make_blobs
from sklearn.decomposition import PCA

X, y = make_blobs(n_samples=100_000_000, n_features=3, centers=[[3, 3, 3], [0, 0, 0], [1, 1, 1], [2, 2, 2]], cluster_std=[0.2, 0.1, 0.2, 0.2], random_state=9)
pca = PCA(n_components=3)
pca.fit(X)
print(pca.explained_variance_ratio_)

from mars.learn.datasets import make_blobs
from mars.learn.decomposition import PCA

X, y = make_blobs(n_samples=100_000_000, n_features=3, centers=[[3, 3, 3], [0, 0, 0], [1, 1, 1], [2, 2, 2]], cluster_std=[0.2, 0.1, 0.2, 0.2], random_state=9)
pca = PCA(n_components=3)
pca.fit(X)
print(pca.explained_variance_ratio_)

Mars 远程

Mars 远程功能允许用户并行执行函数调用。相比传统的函数调用,它能够显著提高执行效率:

import numpy as np
import mars.remote as mr

def calc_chunk(n, i):
    rs = np.random.RandomState(i)
    a = rs.uniform(-1, 1, size=(n, 2))
    return (np.linalg.norm(a, axis=1) < 1).sum()

N = 200_000_000
n = 10_000_000

fs = [mr.spawn(calc_chunk, args=(n, i)) for i in range(N // n)]
pi = mr.spawn(sum, args=(fs,))
print(pi.execute().fetch())

DASK on Mars

Mars 也支持在 DASK 框架上的集成,可以为在分布式数据分析场景中提供优秀的计算能力。

交互模式

Mars 支持交互模式,使其更加易于开发和调试。用户可以通过设置选项来启用交互模式:

from mars.config import options
options.eager_mode = True

与 Ray 的集成

Mars 深度集成了 Ray,可以在 Ray 上高效运行,并与基于 Ray 的机器学习和分布式系统生态无缝交互。

灵活的扩展能力

Mars 可以在单机上运行,也可以扩展至具有数千台机器的集群,使得在不同规模的环境下都能保持优良的性能和数据处理能力。

裸机部署

在裸机集群中,Mars 可以通过启动不同组件来实现集群的扩展。用户可以选择某个节点作为管理节点,其它节点为工作节点,通过启动命令自定义节点角色。

Kubernetes 和 Yarn 部署

Mars 支持通过 Kubernetes 和 Yarn 进行部署,适用于云原生和大数据处理环境。

如何参与

  • 阅读开发指南,加入 Slack 工作群(Mars Computing)或邮件列表(发送邮件至 mars-dev@googlegroups.com)。
  • 感兴趣的开发者可以通过提交 GitHub issue 或 pull requests 来贡献代码。

感谢您的关注与贡献!

项目侧边栏1项目侧边栏2
推荐项目
Project Cover

豆包MarsCode

豆包 MarsCode 是一款革命性的编程助手,通过AI技术提供代码补全、单测生成、代码解释和智能问答等功能,支持100+编程语言,与主流编辑器无缝集成,显著提升开发效率和代码质量。

Project Cover

AI写歌

Suno AI是一个革命性的AI音乐创作平台,能在短短30秒内帮助用户创作出一首完整的歌曲。无论是寻找创作灵感还是需要快速制作音乐,Suno AI都是音乐爱好者和专业人士的理想选择。

Project Cover

白日梦AI

白日梦AI提供专注于AI视频生成的多样化功能,包括文生视频、动态画面和形象生成等,帮助用户快速上手,创造专业级内容。

Project Cover

有言AI

有言平台提供一站式AIGC视频创作解决方案,通过智能技术简化视频制作流程。无论是企业宣传还是个人分享,有言都能帮助用户快速、轻松地制作出专业级别的视频内容。

Project Cover

Kimi

Kimi AI助手提供多语言对话支持,能够阅读和理解用户上传的文件内容,解析网页信息,并结合搜索结果为用户提供详尽的答案。无论是日常咨询还是专业问题,Kimi都能以友好、专业的方式提供帮助。

Project Cover

讯飞绘镜

讯飞绘镜是一个支持从创意到完整视频创作的智能平台,用户可以快速生成视频素材并创作独特的音乐视频和故事。平台提供多样化的主题和精选作品,帮助用户探索创意灵感。

Project Cover

讯飞文书

讯飞文书依托讯飞星火大模型,为文书写作者提供从素材筹备到稿件撰写及审稿的全程支持。通过录音智记和以稿写稿等功能,满足事务性工作的高频需求,帮助撰稿人节省精力,提高效率,优化工作与生活。

Project Cover

阿里绘蛙

绘蛙是阿里巴巴集团推出的革命性AI电商营销平台。利用尖端人工智能技术,为商家提供一键生成商品图和营销文案的服务,显著提升内容创作效率和营销效果。适用于淘宝、天猫等电商平台,让商品第一时间被种草。

Project Cover

AIWritePaper论文写作

AIWritePaper论文写作是一站式AI论文写作辅助工具,简化了选题、文献检索至论文撰写的整个过程。通过简单设定,平台可快速生成高质量论文大纲和全文,配合图表、参考文献等一应俱全,同时提供开题报告和答辩PPT等增值服务,保障数据安全,有效提升写作效率和论文质量。

投诉举报邮箱: service@vectorlightyear.com
@2024 懂AI·鲁ICP备2024100362号-6·鲁公网安备37021002001498号