Add Bedrock support

[viveksilimkhan1]

Description

Add Bedrock Titan and Jurassic support

Corresponding documentation PR

(#338)

Types of change

New feature

Checklist

• I confirm that I have the right to submit this contribution under the project's MIT license.
• I ran all tests in tests and usage_examples/tests, and all new and existing tests passed. This includes
  • all external tests (i. e. pytest ran with --external)
  • all tests requiring a GPU (i. e. pytest ran with --gpu)
• My changes don't require a change to the documentation, or if they do, I've added all required information.

[viveksilimkhan1]

Hi @rmitsch, I have made the changes, please review the code.

[rmitsch]

Sorry for not getting back to you about this yet, @viveksilimkhan1. I have this on my backlog!

[tinutj]

Hello @rmitsch! Can we expect an update on this anytime soon? Thank you!

[zendesk-klross]

Hello, is anything blocking this PR? We also have a strong use case for Bedrock support. Thanks!