naren-1219
/

test-pt-to-jax

Model card Files Files and versions

YAML Metadata Warning:empty or missing yaml metadata in repo card

Check out the documentation for more information.

test-pt-to-jax

PyTorch → JAX/Flax weight conversion demo by Entropy Space.

Conversion rules

PyTorch	Flax	Notes
`Linear.weight [out,in]`	`Dense.kernel [in,out]`	Transposed
`LayerNorm.weight [F]`	`LayerNorm.scale [F]`	Renamed
`Conv.weight [out,in,H,W]`	`Conv.kernel [H,W,in,out]`	Transposed

License

Apache 2.0

Downloads last month: 1

Inference Providers NEW

This model isn't deployed by any Inference Provider. 🙋 Ask for provider support