Initial Cleanup

[siddharth-joshi]

Please do as follows

• Please use requirements.txt for dependency and versioning management
• Update your gitignore to not track files like .DS_Store and build artifacts (e.g., .egg files/contents/folders)
• Remove dead code (e.g., custom_learning.py has dead code for RTRL)
• Run things through a linter, feel free to use the same settings as Flax (Pylint https://github.com/google/flax/blob/main/pylintrc )
• Please set up a set of unit tests for modules (it will make extending your code easier)