You can store the company ID in app_metadata. Using a rule, you can return it in the login token. If you need to find all users from a certain company, you can use the User Search API to search for that particular field in app_metadata.
(One small detail: the User Search API can return a maximum of 1000 users, so keep that in mind whilst performing a search)