Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Add to/from E8M0 scale MX format conversion. #131

Merged
merged 3 commits into from
Aug 12, 2024
Merged

Add to/from E8M0 scale MX format conversion. #131

merged 3 commits into from
Aug 12, 2024

Commits on Aug 12, 2024

  1. Add to/from E8M0 scale MX format conversion. 

    Implementation using bitmasking & shifting, so hopefully decently fast!
    @balancap
    balancap committed Aug 12, 2024
    Configuration menu
    Copy the full SHA
    9f00286 View commit details
    Browse the repository at this point in the history

  2. Pinning JAX to <0.4.31 until sharding parameter bug is solved.

    @balancap
    balancap committed Aug 12, 2024
    Configuration menu
    Copy the full SHA
    759e9e5 View commit details
    Browse the repository at this point in the history

  3. Fixing backward compatibility with JAX 0.3.16. 

    `ml_dtypes` and JAX bfloat16 dtypes not equivalent older JAX versions.
    @balancap
    balancap committed Aug 12, 2024
    Configuration menu
    Copy the full SHA
    32c0510 View commit details
    Browse the repository at this point in the history