[trainer] apex fixes and tests
This PR:

 * [x] fixes a bug in with `fp16_backend` apex (`is_apex_available` + `amp` weren't getting imported w/ pt>=1.6)
 * [x] adds test
 * [x] adds a logger info on which fp16 backend will be used

@sgugger 