The second step computes the remaining pointwise operations.
we compile all point-wise operations into a single fused CUDA kernel and parallelize the computation across each dimension of the hidden state.