Improve export.* APIs

[james77777778]

Fix #27

[codecov]

Codecov Report

Attention: 9 lines in your changes are missing coverage. Please review.

Comparison is base (4ccfee4) 88.49% compared to head (b522fe6) 88.05%.

Files	Patch %	Lines
kimm/export/export_onnx.py	10.00%	9 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             main      #29      +/-   ##
==========================================
- Coverage   88.49%   88.05%   -0.44%     
==========================================
  Files          26       27       +1     
  Lines        2399     2378      -21     
==========================================
- Hits         2123     2094      -29     
- Misses        276      284       +8     

Flag	Coverage Δ
kimm	88.05% <30.76%> (-0.44%)	⬇️
kimm-jax	86.71% <23.07%> (+0.75%)	⬆️
kimm-numpy	86.41% <23.07%> (+0.75%)	⬆️
kimm-tensorflow	87.80% <30.76%> (-0.45%)	⬇️
kimm-torch	86.71% <23.07%> (+0.75%)	⬆️

Flags with carried forward coverage won't be shown. Click here to find out more.

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