Great! It’s a bit disappointing but our present implementation imposes floatX to many parts of distributions, models and step methods, so there’s not so much freedom of movement to choose arbitrary dtypes for different parts of the model
Great! It’s a bit disappointing but our present implementation imposes floatX to many parts of distributions, models and step methods, so there’s not so much freedom of movement to choose arbitrary dtypes for different parts of the model