This pull request improves the CPU fallback mechanism for Mac systems, addresses compatibility issues with JAX versions, and updates the project's README instructions for installation, inference and training.
Changes Made
Modified the CPU fallback logic to handle specific Mac OS configurations.
Fixed compatibility issues with JAX, updating the implementation to support the latest version.
Updated the README with clear instructions for installation and usage.
This pull request improves the CPU fallback mechanism for Mac systems, addresses compatibility issues with JAX versions, and updates the project's README instructions for installation, inference and training.
Changes Made
Related Issues