As soon as you can get a numpy implementation of your forward function, it should work. As said in the documentation there are limits to the numpy functions we support, however.
As soon as you can get a numpy implementation of your forward function, it should work. As said in the documentation there are limits to the numpy functions we support, however.