Support BAdam Optimizer

[bratao]

⚠️ Please check that this feature request hasn't been suggested before.

• I searched previous Ideas in Discussions didn't find any similar feature requests.
• I searched previous Issues didn't find any similar feature requests.

🔖 Feature description

The https://github.com/Ledzy/BAdam it a optimizer that promises a fft with a fraction of the needed memory. It was recently updated to support deepspeed-3

✔️ Solution

Implement https://github.com/Ledzy/BAdam

❓ Alternatives

No response

📝 Additional Context

No response

Acknowledgements

• My issue title is concise, descriptive, and in title casing.
• I have searched the existing issues to make sure this feature has not been requested yet.
• I have provided enough information for the maintainers to understand and evaluate this request.