dev(hansug): add support for torch integration

[HansBug]

Description

import numpy as np
import torch

from treevalue import FastTreeValue, register_treevalue_class


class MyTreeValue(FastTreeValue):
    pass


register_treevalue_class(MyTreeValue)

data = {
    'a': np.random.randint(0, 10, (2, 3)),
    'b': {
        'x': 233,
        'y': np.random.randn(2, 3)
    }
}
t = MyTreeValue(data)

if __name__ == '__main__':
    print(t)

    v1, spec = torch.utils._pytree.tree_flatten(t)
    print(v1)
    print(spec)

    tt = torch.utils._pytree.tree_unflatten(v1, spec)
    print(tt)

Check List

• merge the latest version source branch/repo, and resolve all the conflicts
• pass style check
• pass all the tests

[codecov]

Codecov Report

Merging #79 (adc645e) into main (ac90658) will increase coverage by 0.09%.
The diff coverage is 100.00%.

@@            Coverage Diff             @@
##             main      #79      +/-   ##
==========================================
+ Coverage   98.95%   99.04%   +0.09%     
==========================================
  Files          39       42       +3     
  Lines        2587     2624      +37     
==========================================
+ Hits         2560     2599      +39     
+ Misses         27       25       -2     

Flag	Coverage Δ
unittests	99.04% <100.00%> (+0.09%)	⬆️

Flags with carried forward coverage won't be shown. Click here to find out more.

Impacted Files	Coverage Δ
treevalue/tree/integration/__init__.py	100.00% <100.00%> (ø)
treevalue/tree/integration/base.pyx	100.00% <100.00%> (ø)
treevalue/tree/integration/cjax.pyx	100.00% <100.00%> (+5.55%)	⬆️
treevalue/tree/integration/ctorch.pyx	100.00% <100.00%> (ø)
treevalue/tree/integration/torch.py	100.00% <100.00%> (ø)
treevalue/tree/integration/jax.py	100.00% <0.00%> (+7.14%)	⬆️

Help us with your feedback. Take ten seconds to tell us how you rate us. Have a feature suggestion? Share it here.