[ax] Add badge for gem version