feat: update implementation of Toxicity to use Transform-based approach

[danielezhu]

Description of changes: Update Toxicity implementation and remove last traces of use_ray from code.

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.