Format and streamline style

24 jobs for nifty2jax_parallel_mapping in 41 minutes and 30 seconds (queued for 6 seconds)