tracel-ai / burn

Burn is a new comprehensive dynamic Deep Learning Framework built using Rust with extreme flexibility, compute efficiency and portability as its primary goals.
https://burn.dev
Apache License 2.0
9.01k stars 446 forks source link

Better feature flags for backends. #2390

Open nathanielsimard opened 1 month ago

nathanielsimard commented 1 month ago

I think we should revisit the name of our feature flags and backends. To make things simple we should have:

Some feature flags are meant to customize a backend, they should be specific for that backend:

Essentially, we're making backends made with cubecl first class, meaning we don't need to prefix things with cubecl like cubecl-vulkan, cubecl-cuda or cubecl-fusion, but for the third-party backends, I think it's more flexible, since we won't have feature flags that may clash with our own. In that sense, we may also rename burn-jit to burn-cubecl, since it's pretty much the backend built on top of cubecl.

antimora commented 4 weeks ago

Just to be clear these feature flags are part of the meta package burn?