Also adds an inplace version And fixed the jax code to be equivalent to the Aesara implementation
拖放文件到此处或者 点击上传
