Merge pull request #5 from shyamsn97/notebook-fix

fixing notebooks

bump

Merge pull request #3 from shyamsn97/refactoring_models

refactored and made it easier for inheritance

remove functorch from reqs

update version

updating functional param utils

update generate_params fn

update version

Merge pull request #2 from shyamsn97/refactor

added generate_params_kwargs so that its easy to pass in kwargs that are only meant for generate_params method

updating reinforce notebooks

updating forward method to be more pytorch-y

updated pypi badge