The federal budget will provide Western Australia with almost $21 billion, including $9.5 billion in its share of GST.