enable to set bf16 precision in brainstate.environ module
#274
| Job | Run time |
|---|---|
| 2m 16s | |
| 2m 8s | |
| 3m 11s | |
| 2m 30s | |
| 2m 6s | |
| 2m 27s | |
| 3m 21s | |
| 1m 59s | |
| 2m 32s | |
| 3m 17s | |
| 3m 8s | |
| 28m 55s |