Test Suite for layers, losses

[oke-aditya]

• Adding testing for losses and layers

Plan for layers

• Check that we can use these modules.
• Check values of known cases.

Since layers are bit generic,

Plan for losses.

• Check that we can use these modules
• Check values for known cases
• Loss functions must be differentiable, we should be able to compute grads.
• Check both functional and non functional scenarios.

Currently we shouldn't use these loss functions and fit our models, as these possibilities are endless and would mean incredible increase in CI time (which is already very high)