fayalalebrun

import unittest
import jax
import jax.numpy as jnp
from functools import partial

def decompose(x: jnp.float32) -> tuple[jnp.int32, jnp.int32, jnp.int32]: 
    """decomposes a float32 into negative, exponent, and significand"""
    negative = x < 0
    n = jnp.abs(x).view(jnp.int32)
    exponent = (n >> 23) - 127