support linux aarch64

[AlbertZhangHIT]

System information

• Alpa version:
• Are you willing to contribute it (Yes/No): Yes

Describe the new feature and the current behavior/state
Alpa currently can not support linux aarch64 because jax doesn't support linux aarch64 natively.

Will this change the current API? How?
Not sure.

Describe alternatives you've considered
Supporting jax for linux aarch64 platform.

Additional context