Yu-Hang "Maxin" Tang
Yu-Hang "Maxin" Tang
``` py #!/usr/bin/env python # -*- coding: utf-8 -*- from funfact.lang._tsrex import TsrEx class TensorNetwork: def __init__(self): pass def add_node(self, *tensor_spec_args, **tensor_spec_kwargs): pass def add_edge(self, e: TsrEx): pass def contract(self,...
**Is your feature request related to a problem? Please describe.** The current per-tensor conditions are not applicable if we want to express conditions that involve more than one tensor. **Describe...
Path optimization should only happen between commutative contractions with the same precedence. If connected non-commutative operations share the same user-override precedence, a warning should be raised.
Hi @pghysels, @liuyangzhuan, and @xiaoyeli, Following up on our discussion, we would like to explore the possibility of subclassing [GaussianProcessRegressor](https://scikit-learn.org/stable/modules/generated/sklearn.gaussian_process.GaussianProcessRegressor.html) from scikit-learn. The definition of the regressor is [here](https://github.com/scikit-learn/scikit-learn/blob/master/sklearn/gaussian_process/_gpr.py). The...
**Describe the bug** If the same tensor shows up more than once in a tensor expression, then it cannot be correctly factorized. Related to #210, #228, #232. **To Reproduce** ```...
Together with https://github.com/google/jax/pull/15205, this should eliminate the need to apply the `build_jaxlib/update_build_scripts.patch` each time we rebase. Note the `--editable` option will only be usable once alpa is rebased onto JAX...
This is still a work-in-progress. - [ ] Migrate from tensorflow to OpenXLA - [ ] migrate all delta between tensorflow-alpa and tensorflow upstream to [xla-alpa](https://github.com/yhtang/xla-alpa) - [ ] build...
[JEstimator 0.3.3 on PyPI ](https://pypi.org/project/jestimator/0.3.3/) is incompatible with the latest JAX due to the relocation of the `PartitionSpec` endpoint: ``` File "/opt/t5x/t5x/optimizers.py", line 38, in from jestimator import amos File...
**Describe the bug** If a data class is created using the function `dataclasses.make_dataclass` instead of the `@dataclasses.dataclass` decorator, then the docstring (and even the entire node attribute data) is empty....