Improve weights conversion to ONNX

[FynnBe]

Our ONNX converters could be improved.

To convert from pytorch ONNX dynamo looks promising.

[oeway]

I would also like to add that if you convert to ONNX, please set the input shape, on the x, y dimension to -1, currently it causing issue for the BioEngine to support arbitrary input size.