为JAX中抽象数组类型的多重分派准备的quaxed库
项目描述
quaxed
预-Quaxify
'ed JAX
Quaxed
使用 jax 库(通过 quax
)来封装,以便使用这些库与自定义类似数组的对象一起使用,而不仅仅是 jax 数组。
安装
pip install quaxed
文档
快速开始
要了解 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')
开发
我们欢迎贡献!
引用
如果您认为这个库很有用,并希望支持科学社区低级实用库的开发和维护,请考虑引用这项工作。
项目详情
下载文件
下载您平台上的文件。如果您不确定选择哪个,请了解更多关于 安装包 的信息。
源分发
quaxed-0.6.4.tar.gz (57.1 kB 查看散列)
构建分发
quaxed-0.6.4-py3-none-any.whl (35.8 kB 查看散列)
关闭
quaxed-0.6.4.tar.gz 的散列
算法 | 散列摘要 | |
---|---|---|
SHA256 | b66122a702af0c43f4a5565b4fd530bcd93911af3aa25eed9a8ff8f88c9d0326 |
|
MD5 | 0a51b5d7a2e2bff07836f9cee6abdaaf |
|
BLAKE2b-256 | cd3caa18f0b95b3c99ac892f96fb99ed438cb6aaabf352c2fc71ddda84a29788 |
关闭
quaxed-0.6.4-py3-none-any.whl 的散列
算法 | 散列摘要 | |
---|---|---|
SHA256 | 850af2fe1bebb48b687bfcd491b9ce17fadabd83fa0f869a18e86f8b5d41452c |
|
MD5 | e2b8ebca8457bd67d3e18691004c491c |
|
BLAKE2b-256 | e2118074e2f4e875f304992d7de4bd2d50193ccd9efbc4002f2b9d8033affd12 |