To model binomial count data for VAE


I am new to VAE. I understand VAE can model the input data as Gaussian distribution, but
I have a large number of count data that follow binomial distributions. May I ask if it is possible to use PyTorch to implement a VAE of binomial inputs? Thank you.