test-pt-to-jax – AI Model by naren-1219 | AlphaNeural AI

You can deploy this model and start earning money today!

naren-1219/
test-pt-to-jax

like

0

Views

No views yet

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