Code Refactor for Final Release

[pomonam]

This PR implements a comprehensive code refactoring to prepare for its final release. The changes focus on improving code quality, performance, and maintainability.

• Improved code readability and structure
• Optimized performance and memory usage
• Updated all example files to reflect recent changes
• Revised documentation to accurately represent the current codebase
• Expanded pytest coverage to ensure code correctness (now, we have the identical set of tests with the internal implementation)

Some TODOs:

• Enhance AMP (Automatic Mixed Precision) logic.
• Implement regression tests to verify consistency with the internal version

[pomonam]

Per token IF computation (#10) is also implemented.