Downloads · 30 days
6
11% of all-time downloads
test-pt-to-jax
naren-1219/test-pt-to-jax
test-pt-to-jax is a machine learning model from naren-1219. Use it for the machine learning task on the model card, and read the license before you ship it in a product.
PyTorch → JAX/Flax weight conversion demo by Entropy Space.
All-time downloads
54
Public
Parameters
33.3K
134 KB on disk
Likes
0
Public
Repo makeup
Click a slice to open those files.
.safetensors134 KB · 98%
At a glance
- Access
- Public
- Created
- Mar 18, 2026
- Updated
- Mar 18, 2026
- SHA
- 4ea2d1da
Model card
From the Hugging Face model README
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
- Created
- Mar 18, 2026
- Updated
- Mar 18, 2026