Remove lax_ prefixes from jax/_src/lax filenames, since they aren't needed any longer to avoid name conflicts.