Jax/PyTorch Backend

Open benjaminpope opened this issue 2 months ago • 0 comments

Hi Adina,

I hope you're well! As noted in Issue #32, recent versions of tensorflow don't play well with stella, and this has made it a bit of a challenge for some students to use.

As an experiment, I sat down with the GPT5 Copilot and tried agent mode for the first time... I hadn't tried it before, and it was actually a lot of work to use. Blame me for any bugs in there.

Anyway - as far as I can tell, stella now works with Jax and PyTorch backends and python 3.12. As far as I can tell on light curves I've already looked at, I get the same answers as before - but I don't have a working install of old stella on this machine, so I can't rigorously compare them. I have tested all the functions I have used but not the NN training, and the performance seems good. I've added a few convenience functions.

Let me know what you think!

All the best,

Ben

Nov 04 '25 15:11 benjaminpope