v0.4.0: API change (onnx package refactored); Many new ops and models support; Fused ops;
#72
janpfeifer
announced in
Announcements
Replies: 0 comments
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
-
This release include the new v0.4.0 and v0.3.[56] (never actually released) versions.
v0.4.0: More ops and fused ops; Added support for model Florence-2;
onnxpackage refactored.onnx: split implementation intointernal/onnxgomlx/...,onnxis now just a public API.onnx.Modelis now an interface (it was a pointer to an object)and the constructors reside in
onnx/parser.Modoperator: Supports both fmod=1 (C-style, sign follows dividend) and fmod=0 (Python-style, sign follows divisor) with broadcasting and dtype promotiononnxImplicitFloatPromotionfor float-only ops (Sqrt, Exp, etc.).Concatdtype alignment: When dtype promotion is enabled, all Concat operands are cast to the first operand's dtype, preserving Int64 for shape/index tensors.isVariableConstantloosening: Float variables with "const" in the name are now accepted as materializable constants (needed when Concat dtype promotion casts Float32 constants to Int64).convertIf()rework: Uses GoMLX's native If with closures instead of the Where-based approach.v0.3.5: New ops for various models support (Gemma, Snowflake,CLAP, etc); Many fixes and improvements.
isZeroInitializerto handle tensors with zero-sized dimensions (e.g.,[batchSize, 0]).ReduceL2,SimplifiedLayerNormalization(RMSNorm),RotaryEmbedding,MultiHeadAttention.in ONNX specification, but some Pytorch models do this. See
Model.AllowDTypePromotion().This discussion was created from the release v0.4.0: API change (`onnx` package refactored); Many new ops and models support; Fused ops;.
Beta Was this translation helpful? Give feedback.
All reactions