Jax/PyTorch Backend#34

Open

benjaminpope wants to merge 49 commits intoafeinstein20:mainfrom

benjaminpope:main

Commits on Jun 11, 2024

Update requirements.txt
benjaminpope
authored

Commits on Nov 4, 2025

Merge pull request #1 from afeinstein20/main
benjaminpope
authored
Migrate to Keras Core (JAX); drop TensorFlow; use .keras; update CI/docs; add scripts; bump Python
benjaminpope
committed
keras fixes
benjaminpope
committed
tests: use bundled model in test_predict; register downloads marker; remove dummy model
benjaminpope
committed
package: move model .keras files into stella/data for packaging
benjaminpope
committed
package: include stella/data/*.keras; tests: load model via importlib.resources
benjaminpope
committed
feat(models): add stella.models helpers and use in tests/docs; build: migrate to pyproject.toml and include stella/data/*.keras
benjaminpope
committed
feat(models): expose stella.models.models list; docs: use packaged models in shortest demo
benjaminpope
committed
feat(predict): add per-window progress with chunked inference; pipe through in pipeline; build: add optional deps for jax/torch in pyproject
benjaminpope
committed
fix(mark_flares): return ragged groups as object-dtype array to avoid NumPy 1.24+ ValueError
benjaminpope
committed
feat(backends): add require_backend(); chore: migrate pytest config to pyproject and remove setup.*
benjaminpope
committed
fix(mark_flares): cast per-lightcurve arrays to float to avoid object dtype in medfilt/interp1d
benjaminpope
committed
docs(install): document backend extras (jax/jax-mps/torch), KERAS_BACKEND, and stella.require_backend()
benjaminpope
committed
feat(backends): auto-select default Keras backend (prefer JAX then Torch); fix rotations Time/Quantity handling; modernize tqdm imports; remove hard-coded paths and Dropbox refs; docs pipeline uses…
benjaminpope
committed
chore(repo): ignore and untrack downloaded FITS data (mastDownload/) to avoid large artifacts in commits
benjaminpope
committed
chore(repo): stop tracking local notebooks and scripts\n\n- Ignore and in .gitignore\n- Remove both from index with (files remain locally)
benjaminpope
committed
update python version
benjaminpope
committed
ci: install gettext on macOS before setup-python to fix libintl.8.dylib error
benjaminpope
committed
ci(tests): install package in editable mode with dev+jax extras before pytest
benjaminpope
committed
ci: modernize GitHub Actions versions (checkout@v4, setup-python@v5, pages deploy@v4)
benjaminpope
committed
ci: remove legacy npm docs workflow (no package.json); use Sphinx docs.yml instead
benjaminpope
committed
ci: use pip install . instead of setup.py install (pyproject build)
benjaminpope
committed
ci: use native Python arch on macOS-14 (remove architecture: x64)
benjaminpope
committed
ci: unify all workflows to Python 3.12 (CI, tests, docs)
benjaminpope
committed
tests: skip JAX-specific tests when jax is not installed (importorskip)
benjaminpope
committed
docs: fix theme 'style' context and nbsphinx issues in tutorial\n\n- Provide html_context['style'] default in conf.py to avoid Jinja UndefinedError\n- Switch to tqdm.auto in tutorial imports\n- Wra…
benjaminpope
committed
suggested fixes
benjaminpope
committed
minimal torch
benjaminpope
committed
docs(mkdocs): migrate from Sphinx to MkDocs\n\n- Add mkdocs.yml with material theme, mkdocstrings, mkdocs-jupyter\n- Convert key pages to Markdown and include notebooks in nav\n- Update docs workfl…
benjaminpope
committed
torch is still a problem on mac
benjaminpope
committed
docs(mkdocs): port Sphinx content to Markdown, add logo asset, trigger on mkdocs.yml, ignore site/
benjaminpope
committed
stella logo
benjaminpope
committed
docs: add Markdown versions of Shortest Demo and Other Features; update MkDocs nav
benjaminpope
committed
docs: replace Tutorial with Markdown page; ci: run Torch tests on Linux to avoid macOS MPS OOM
benjaminpope
committed
remove greetings
benjaminpope
committed
docs(ci): run docs workflow on any push/PR to main, feat/**, fix/** (remove paths filter)
benjaminpope
committed
Merge pull request #2 from benjaminpope/feat/keras-jax
benjaminpope
authored
fix landing page
benjaminpope
committed
Merge pull request #4 from benjaminpope/feat/keras-jax
benjaminpope
authored
fix links
benjaminpope
committed
Merge pull request #5 from benjaminpope/feat/keras-jax
benjaminpope
authored

Commits on Nov 5, 2025

docs: render original notebooks in nav (mkdocs-jupyter)
benjaminpope
committed

Commits on Nov 6, 2025