Merge `is_prime` std and no-std implementations

Currently, we have two different implementations for `is_prime` (one with `std` and another without). We should see if we can merge both without impacting performance (the `no-std` one runs ~10% faster than the `std`)