---
license: mit
library_name: jax
tags:
- jax
- equinox
- lewm
- world-model
- safetensors
---

# LeWM cube JAX checkpoint

JAX-native Safetensors export of `quentinll/lewm-cube` for
[le-wm-jax](https://github.com/davidhidary/le-wm-jax).

Files:

```text
config.json
model.safetensors
```

Load:

```python
from le_wm_jax.checkpoint import load_lewm_jax

model = load_lewm_jax("DavidHidary/lewm-cube-jax")
```

The original PyTorch checkpoint is from the LeWM release by Lucas Maes and collaborators.