Skip to content

Releases

StaleyNET v0.1.2

Choose a tag to compare

@arpanpathak arpanpathak released this 04 Nov 08:14
b1c8ac4

StaleyNET v0.1.2

Highlights

  • A crate page for newcomers. staleynet-audio 0.1.2 is on crates.io.
    • Its page explains decoding, resampling, the log-mel spectrogram and Griffin-Lim reconstruction by intuition first.
    • It has five examples, and every one runs as a documentation test.
    • It links to the paper, the book chapters and the background literature.
  • GPU kernels organised by topic (#35, #36).
    • There is one module per topic: matrix products, attention, loss, optimizer, element-wise operations, normalization, embeddings and signal processing.
    • Long argument lists are replaced by small documented structures.
    • Writes go through two bounds-checked helpers.
  • Current cuda-oxide (#34). The AGX Thor builds against cuda-oxide's main branch on Rust 1.99.1, and the Orin NX keeps its older revision through a local override. One code path serves both.
  • Revised paper (#38).
    • The text is plainer and more modest, and every quantity carries its unit.
    • Procedures state their inputs and outputs.
    • A new appendix covers hardware (including the AGX Thor's 128 GB), benchmarks and the tokenizer study.
    • The book's GPU chapter describes the new kernel layout.
  • Safer remote sync (#37). deploy/thor/sync.sh no longer deletes caches that exist only on the remote machine.
  • New artifacts.
    • The private Hugging Face repositories now hold the lyrics-thor, lyrics-thor-long and pq-large runs. pq-large is a 42.9 M-parameter model on the band-group tokenizer, uploaded as a snapshot at step 7,000.
    • They also hold the band-group token caches.
    • artifacts/ records their checksums and metadata.

Install the audio crate

cargo add staleynet-audio

Crates.io status

staleynet-audio is published. The other crates remain source-only because they depend on cuda-oxide's cuda-device and cuda-host, which are distributed only through Git, and crates.io does not accept Git dependencies.

Verification

  • CI: rustfmt, clippy with warnings denied, tests, cargo-deny and the audio coverage gate pass.
  • Audio crate: 40 unit tests and 5 documentation tests pass.
  • GPU crates, on both the Jetson Orin NX (cuda-oxide b7fe543) and the Jetson AGX Thor (cuda-oxide 2ffc19b):
    • staleynet-cuda: 31 unit and 9 integration tests pass;
    • staleynet-model: 32 and 6 pass;
    • staleynet-data: 27 and 3 pass.

StaleyNET v0.1.1

Choose a tag to compare

@arpanpathak arpanpathak released this 04 Nov 07:08
e2aa933

StaleyNET v0.1.1

Initial tagged release of the rebuilt StaleyNET repository.

Highlights

  • Clean 33-PR history, rebuilt module by module.
  • 219 files covering code, paper, book, studio, configs, and artifact metadata.
  • 171 tests pass, including 18 GPU integration tests on a Jetson Orin.
  • CI covers rustfmt, clippy with warnings denied, tests, cargo-deny, and the 99 percent audio coverage gate.
  • staleynet-audio 0.1.1 is published on crates.io with a crate-specific README.
  • The private Hugging Face model and dataset cards were rewritten for this release.

Install the audio crate

cargo add staleynet-audio

Repository layout

  • crates/staleynet-audio: CPU decoding, resampling, log-mel, Griffin-Lim, WAV.
  • crates/staleynet-cuda: cuda-oxide kernels and shape-checked operations.
  • crates/staleynet-model: Transformer, optimizer, decoding, checkpoints.
  • crates/staleynet-data: catalogues, cache store, RVQ tokenizer, sampler.
  • crates/staleynet-train: training loop and checkpoint policy.
  • crates/staleynet-cli: the staleynet command.
  • paper/: LaTeX sources and compiled PDFs.
  • book/: mdBook source and figures.
  • studio/: browser studio for generation and recording.
  • artifacts/: run metadata, checksums, and Hugging Face manifest.

Crates.io status

staleynet-audio is published. The GPU crates remain source-only for now because they depend on cuda-oxide through git revisions, and crates.io rejects git dependencies. They can be packaged once cuda-oxide publishes the host and device crates, or on a dedicated packaging branch.

Verification

  • cargo fmt --all -- --check
  • cargo clippy --workspace --all-targets -- -D warnings
  • cargo deny check
  • cargo test -p staleynet-audio
  • cargo oxide test -- --release --workspace on the Jetson Orin
  • Audio line coverage: 99.10 percent

Links

License

Code is Apache-2.0. Paper and book are CC BY 4.0. Checkpoints and derived data are CC BY-SA 4.0 subject to the upstream dataset terms.