Skip to main content

CASTOR

CI Documentation Status PyPI License

CASTOR is a transformer-based pipeline for gravitational-wave detection, training, inference, and sensitivity evaluation.

CASTOR is under active development. Version 0.1.0 packages the analysis workflow described in the accompanying paper.

Installation

Choose and install the appropriate CPU, CUDA, or ROCm build of PyTorch for your platform before installing CASTOR. This prevents a CPU-only Linux environment from downloading an unintended CUDA runtime. Then install the released package with:

python -m pip install castor-gw

For an editable development installation, install from a clone:

git clone https://github.com/chayanchatterjee/castor.git
cd castor
python -m pip install -e .

Install the scientific extras needed by a workflow with:

python -m pip install -e ".[training]"   # ML4GW online training
python -m pip install -e ".[evaluation]" # PyCBC whitening
python -m pip install -e ".[plotting]"   # plotting dependencies

Verify the installed commands with:

castor-train --help
castor-eval --help

See the documentation for the training, evaluation, cluster, data-format, and reproducibility guides.

Reproducibility

The paper workflow uses 10--50 solar-mass component masses, 500 epochs, global seed 2026, deterministic validation, raw checkpoint weights, and a factor of 32 that reconciles the PyCBC and ML4GW whitening conventions. Checkpoints and evaluation caches record the configuration and input identities required to reject incompatible reuse.

License

CASTOR is distributed under the Apache License 2.0.

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

castor_gw-0.1.0.tar.gz (61.7 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

castor_gw-0.1.0-py3-none-any.whl (53.0 kB view details)

Uploaded Python 3

File details

Details for the file castor_gw-0.1.0.tar.gz.

File metadata

  • Download URL: castor_gw-0.1.0.tar.gz
  • Upload date:
  • Size: 61.7 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/7.0.0 CPython/3.13.14

File hashes

Hashes for castor_gw-0.1.0.tar.gz
Algorithm Hash digest
SHA256 f8343f604f024b2052e2daa1d6611682de0f9a56ffabc32c1fe61908dce73325
MD5 89ba6652cfc73011c80bdd0073c007bb
BLAKE2b-256 206dfde18dfb15d8339c24dc332b039372c60443746791bb9d787a52ec68bb50

See more details on using hashes here.

Provenance

The following attestation bundles were made for castor_gw-0.1.0.tar.gz:

Publisher: release.yml on chayanchatterjee/castor

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file castor_gw-0.1.0-py3-none-any.whl.

File metadata

  • Download URL: castor_gw-0.1.0-py3-none-any.whl
  • Upload date:
  • Size: 53.0 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/7.0.0 CPython/3.13.14

File hashes

Hashes for castor_gw-0.1.0-py3-none-any.whl
Algorithm Hash digest
SHA256 ef7014a4f6f2090e7e775b5b1d52b8a0164843334279767f326a60e95c99f3f6
MD5 8499ed4f3d5ac148a7226b17af8bad1f
BLAKE2b-256 f7561965e32cd4715d9c179fce7ead990d29e48d958a0f151f1e766f3d2b4c53

See more details on using hashes here.

Provenance

The following attestation bundles were made for castor_gw-0.1.0-py3-none-any.whl:

Publisher: release.yml on chayanchatterjee/castor

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Release history Release notifications | RSS feed

This release

0.1.0 This release

2 files

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page