xmmutablemap

JAX兼容的不可变映射

JAX更倾向于不可变对象，但Python和JAX都没有提供不可变字典。😢
此存储库定义了一个轻量级的不可变映射（低于dict），JAX将其视为PyTree。🎉 🕶️

安装

pip install xmmutablemap

文档

xmutablemap提供了类ImmutableMap，它是Python的Mapping ABC的完整实现。如果您使用过dict，那么您已经知道如何使用ImmutableMap！ImmutableMap增加的功能包括1）不可变性（以及相关的优势，如可哈希性）和2）与JAX的兼容性。

from xmmutablemap import ImmutableMap

print(ImmutableMap(a=1, b=2, c=3))
# ImmutableMap({'a': 1, 'b': 2, 'c': 3})

print(ImmutableMap({"a": 1, "b": 2.0, "c": "3"}))
# ImmutableMap({'a': 1, 'b': 2.0, 'c': '3'})

开发

我们欢迎贡献！