Add AMD support

Don't delete this branch after merging (I have a PR open for the original repo)