Add ops w/ default implementation for QTensorOps

[laggui]

Checklist

• Confirmed that run-checks all script has been executed.

Related Issues/PRs

Follow-up to #2025

Changes

• Added float ops as q_* ops in QTensorOps with some default implementations
  • All float ops now match based on the tensor primitive enum (Float vs QFloat) to call the corresponding ops
  • Implemented ops for ndarray and tch backends
• Added quantize_dynamic which computes the quantization parameters at runtime
• Added testgen_quantization test macro to use with backends that support quantization
• Added missing sort_with_indices op to tch backend
• Refactored existing chunk to call the corresponding backend methods (previously always ran the default implementation regardless of backend implementation)

Testing

Added unit tests for all ops.

Note: the tests try to use tensors with floating point values which can be de/quantized without introducing too much quantization error, but the result always depends on the operation (e.g., tensor product of values can grow larger and significantly increase the output tensor range, leading to more de/quantization error on the results).

[codecov]

Codecov Report

Attention: Patch coverage is 88.05777% with 645 lines in your changes missing coverage. Please review.

Project coverage is 85.83%. Comparing base (ccb5b22) to head (d5e6976).
Report is 2 commits behind head on main.

Files with missing lines	Patch %	Lines
crates/burn-tch/src/ops/qtensor.rs	0.00%	196 Missing ⚠️
crates/burn-tensor/src/tensor/ops/qtensor.rs	71.52%	162 Missing ⚠️
crates/burn-autodiff/src/ops/qtensor.rs	0.00%	60 Missing ⚠️
crates/burn-fusion/src/ops/qtensor.rs	0.00%	48 Missing ⚠️
crates/burn-candle/src/ops/qtensor.rs	0.00%	43 Missing ⚠️
crates/burn-jit/src/ops/qtensor.rs	0.00%	43 Missing ⚠️
...es/burn-tensor/src/tests/quantization/ops/chunk.rs	77.87%	25 Missing ⚠️
crates/burn-tensor/src/tensor/api/numeric.rs	89.23%	21 Missing ⚠️
crates/burn-ndarray/src/ops/qtensor.rs	82.41%	16 Missing ⚠️
crates/burn-tensor/src/tensor/api/base.rs	81.70%	15 Missing ⚠️
... and 3 more

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #2125      +/-   ##
==========================================
+ Coverage   85.57%   85.83%   +0.25%     
==========================================
  Files         707      750      +43     
  Lines       89097    94420    +5323     
==========================================
+ Hits        76249    81048    +4799     
- Misses      12848    13372     +524     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

[github-actions]

This PR has been marked as stale because it has not been updated for over a month