Observe this call to `time`:

```
(jax_env) ➜  purejaxrl git:(main) ✗ time python foo.py
python foo.py  11.41s user 0.90s system 109% cpu 11.230 total
```

If I understand correctly, the 11.230 number is 11.230 seconds, but for
some reason it doesn't have "s" after it like 11.41s and 0.90s. I think
that it should.


Thanks,
Ram.