• Jesse Grabowski's avatar
    Cleanup for Optimal Control Ops (#1045) · fffb84c1
    Jesse Grabowski 提交于
    * Blockwise optimal linear control ops
    
    * Add jax rewrite to eliminate `BilinearSolveDiscreteLyapunov`
    
    * set `solve_discrete_lyapunov` method default to bilinear
    
    * Appease mypy
    
    * restore method dispatching
    
    * Use `pt.vectorize` on base `solve_discrete_lyapunov` case
    
    * Apply JAX rewrite before canonicalization
    
    * Improve tests
    
    * Remove useless warning filters
    
    * Fix local_blockwise_alloc rewrite
    
    The rewrite was squeezing too many dimensions of the alloced value, when this didn't have dummy expand dims to the left.
    
    * Fix float32 tests
    
    * Test against complex inputs
    
    * Appease ViPy (Vieira-py type checking)
    
    * Remove condition from `TensorLike` import
    
    * Infer dtype from `node.outputs.type.dtype`
    
    * Remove unused mypy ignore
    
    * Don't manually set dtype of output
    
    Revert change to `_solve_discrete_lyapunov`
    
    * Set dtype of Op outputs
    
    ---------
    Co-authored-by: 's avatarricardoV94 <ricardo.vieira1994@gmail.com>
    fffb84c1
名称
最后提交
最后更新
.github 正在载入提交数据...
doc 正在载入提交数据...
pytensor 正在载入提交数据...
scripts 正在载入提交数据...
tests 正在载入提交数据...
.gitattributes 正在载入提交数据...
.gitignore 正在载入提交数据...
.pre-commit-config.yaml 正在载入提交数据...
.readthedocs.yaml 正在载入提交数据...
CODE_OF_CONDUCT.md 正在载入提交数据...
CONTRIBUTING.md 正在载入提交数据...
GOVERNANCE.md 正在载入提交数据...
LICENSE.txt 正在载入提交数据...
MANIFEST.in 正在载入提交数据...
README.rst 正在载入提交数据...
codecov.yml 正在载入提交数据...
conftest.py 正在载入提交数据...
environment-osx-arm64.yml 正在载入提交数据...
environment.yml 正在载入提交数据...
pyproject.toml 正在载入提交数据...
setup.py 正在载入提交数据...