Create dataclasses to hold frameworks for added flexibility

[sayakpaul]

Not a strong opinion but should torch and tvm-ffi come from a set constant?

Or even better could be to maintain a set of framework where each framework would be a dataclass having a field for its identifier name. But I understand this is probably out of scope for this PR and introduces complexity that I am not if we want.

Originally posted by @sayakpaul in #320

[sayakpaul]

Additional reference: https://github.com/huggingface/kernels/pull/320/changes#r2910189700

[danieldk]

I think this is fixed by #339.