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.