Add Sound Encoder to Cosmos3 (#13911)
* Initial version of sound encoder
Signed-off-by: Maciej Bala <mbala@nvidia.com>
* Answered review comments
Signed-off-by: Maciej Bala <mbala@nvidia.com>
* Reshaped encoder Snake1d
Signed-off-by: Maciej Bala <mbala@nvidia.com>
* Updated test
Signed-off-by: Maciej Bala <mbala@nvidia.com>
* Fixed test
Signed-off-by: Maciej Bala <mbala@nvidia.com>
* Apply style fixes
* Added docstring
Signed-off-by: Maciej Bala <mbala@nvidia.com>
---------
Signed-off-by: Maciej Bala <mbala@nvidia.com>
Co-authored-by: dg845 <58458699+dg845@users.noreply.github.com>
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
