Skip to content
This repository was archived by the owner on Jul 17, 2026. It is now read-only.

Commit fdb2eb3

Browse files
ocg-goodfireclaude
andcommitted
rename: nano_param_decomp_jax -> param_decomp_jax (drop the nano-era name)
Distribution dir + package name (nano-pd-jax -> param-decomp-jax); Makefile targets, launcher paths, docs updated. The torch nano_param_decomp reference impl keeps its name — minimal-reference is what it is. Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
1 parent d7b54ba commit fdb2eb3

82 files changed

Lines changed: 15 additions & 15 deletions

File tree

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

Makefile

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -21,25 +21,25 @@ install-all: install-dev install-app
2121
# really want a from-scratch env.
2222
.PHONY: install-jax
2323
install-jax:
24-
cd nano_param_decomp_jax && ([ -x .venv/bin/python ] || \
24+
cd param_decomp_jax && ([ -x .venv/bin/python ] || \
2525
( ! [ -e .venv ] || mv .venv .venv-stale-$$(date +%s); \
2626
uv venv .venv --python 3.13; rm -rf .venv-stale-* || true )) \
2727
&& uv pip install -p .venv/bin/python -e ../param_decomp_config -e '.[dev]'
2828

2929
.PHONY: install-jax-cuda
3030
install-jax-cuda:
31-
cd nano_param_decomp_jax && ([ -x .venv-cuda/bin/python ] || \
31+
cd param_decomp_jax && ([ -x .venv-cuda/bin/python ] || \
3232
( ! [ -e .venv-cuda ] || mv .venv-cuda .venv-stale-$$(date +%s); \
3333
uv venv .venv-cuda --python 3.13; rm -rf .venv-stale-* || true )) \
3434
&& uv pip install -p .venv-cuda/bin/python -e ../param_decomp_config -e '.[cuda]'
3535

3636
.PHONY: test-jax
3737
test-jax:
38-
cd nano_param_decomp_jax && .venv/bin/python -m pytest jax_single_pool/tests/
38+
cd param_decomp_jax && .venv/bin/python -m pytest jax_single_pool/tests/
3939

4040
.PHONY: check-jax
4141
check-jax:
42-
cd nano_param_decomp_jax && .venv/bin/basedpyright jax_single_pool/
42+
cd param_decomp_jax && .venv/bin/basedpyright jax_single_pool/
4343

4444

4545
.PHONY: app
Lines changed: 1 addition & 1 deletion

nano_param_decomp_jax/jax_single_pool/INTEGRATION.md renamed to param_decomp_jax/jax_single_pool/INTEGRATION.md

Lines changed: 1 addition & 1 deletion

nano_param_decomp_jax/jax_single_pool/LOSS_PARITY_DESIGN.md renamed to param_decomp_jax/jax_single_pool/LOSS_PARITY_DESIGN.md

nano_param_decomp_jax/jax_single_pool/README.md renamed to param_decomp_jax/jax_single_pool/README.md

Lines changed: 1 addition & 1 deletion

0 commit comments

Comments
 (0)