danielward27 / flowjax

https://danielward27.github.io/flowjax/
MIT License
82 stars 10 forks source link

Types #158

Closed danielward27 closed 4 months ago

danielward27 commented 4 months ago

Making parameter type conversion more consistent by converting to float arrays where appropriate. Important to be considerate of float vs int as the type is generally used as a filter for trainable parameters.