- Added a new `detections` attribute to the `AffinityResult` class to store detection objects used in affinity matrix calculations.
- Introduced a `tracking_detections` method to yield pairs of trackings and their corresponding detections, enhancing the functionality of the class.
- Updated type hints for improved clarity and consistency in the class definition.
- Added a new `AffinityResult` class to encapsulate the results of affinity computations, including the affinity matrix, trackings, and their respective indices.
- Introduced a vectorized implementation of `calculate_camera_affinity_matrix_jax` to enhance performance by leveraging JAX's capabilities, replacing the previous double-for-loop approach.
- Updated tests in `test_affinity.py` to include parameterized benchmarks for comparing the performance of the new vectorized method against the naive implementation, ensuring accuracy and efficiency.