跳转到主要内容

为JAX中抽象数组类型的多重分派准备的quaxed库

项目描述

quaxed

预-Quaxify'ed JAX

Quaxed 使用 jax 库(通过 quax)来封装,以便使用这些库与自定义类似数组的对象一起使用,而不仅仅是 jax 数组。

安装

PyPI version PyPI platforms

pip install quaxed

文档

Documentation Status

快速开始

要了解 quax 的工作原理和魔法,请参阅 quax.quaxify教程

使用这个库非常简单

# Import pre-quaxified library
>>> import quaxed.numpy as jnp  # this is quaxify(jax.numpy)

# As an example, let's import an array-ish object
>>> from unxt import Quantity
>>> x = Quantity(2, "km")
>>> jnp.square(w)
Quantity['area'](Array(4, dtype=int64, weak_type=True), unit='km2')

开发

Actions Status Codecov

我们欢迎贡献!

引用

DOI

如果您认为这个库很有用,并希望支持科学社区低级实用库的开发和维护,请考虑引用这项工作。

项目详情


下载文件

下载您平台上的文件。如果您不确定选择哪个,请了解更多关于 安装包 的信息。

源分发

quaxed-0.6.4.tar.gz (57.1 kB 查看散列)

上传时间

构建分发

quaxed-0.6.4-py3-none-any.whl (35.8 kB 查看散列)

上传时间 Python 3

由以下机构支持