Skip to content

Updates for JAX 0.7 #32

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom
Merged
leakec merged 45 commits into from
Aug 4, 2025
Merged

Updates for JAX 0.7 #32

leakec merged 45 commits into from
Aug 4, 2025

Conversation

@leakec
Copy link
Owner

@leakec leakec commented Aug 4, 2025

Updates for JAX 0.7 by updating to use pybind11 rather than swig. This is a rather large update that does the following:

  • Uses pybind11 to build the extension rather than swig.
  • Uses cmake for building
  • Added more build flags to catch more errors in C++ -Werror -Wall -Wextra -pedantic
  • Also, has some typing updates to make types behave a bit better across the board.

Closes #31

leakec added 30 commits July 26, 2025 08:40
@leakec
Working on extension with pybind11
a878d7d
@leakec
Can now install the extension module and stubs.
650c0d2
@leakec
Got this working. Can at least pass the basic test.
6fd3e55
@leakec
Removing files we don't need anymore.
90a7ff4
@leakec
Adding other non-ELM 1-D basis functions.
35c0c24
@leakec
Updating so we can find pybind11 from anywhere on different machines.
f09c06f
@leakec
Adding pybind11 and removing unecessary imports.
42f4904
@leakec
Adding in 1-D ELM functions.
d9956df
@leakec
Fixing classifiers warning.
fc928ec
@leakec
Working on updating ci to build.
3e77389
@leakec
Got nBasisFuncs to compile and pass reg tests. Now, only nELMs remain.
4d66266
@leakec
Adding nELM functions.
464ca6f
@leakec
Updating so unit tests pass.
a34e941
@leakec
Fixing API version to avoid warnings.
76efc4a
@leakec
Formatting with black.
f2838f7
@leakec
Adding pybind11 to requirements for docs.
fbeb5df
@leakec
Removing this file as I don't think we need it anymore.
a04413f
@leakec
Adding pybind11 and mypy.
7ac2f90
@leakec
Fixing types now that we are on 3.10 or greater.
deb9f49
@leakec
Reworkign types and moving BF into utils. No need for separate folder.
f1fd828
@leakec
Trying to install package now in readthedocs since we use cmake.
8f433af
@leakec
Switching runner to ubuntu 24, using python 3.12, and adding cmake fo…
c7a3a05 
…r readthedocs.

Adding ninja.

Adding ninja.
@leakec
Changing minimum version to 3.11.
d172687
@leakec
Changing to BF.cc
40ba76e
@leakec
Fixing input path.
4456a02
@leakec
Formatting C++ code using clang.
f417eba
@leakec
Trying cast to int to make Mac OS happy.
224f118
@leakec
Changes so this builds with pedantic.
df33beb
@leakec
Adding files to manifest for source distribution.
6858833
@leakec
Switching to stringstream, as Mac OS seems to be having issues with s…
07ff93a 
…td::format.
leakec added 15 commits August 2, 2025 16:10
@leakec
Adding other dependencies and removing swig.
55c04b6
@leakec
Removing 3.14 for now.
9611ec3
@leakec
Adding Python version to setup Python to get rid of warnings.
04b1840
@leakec
Updating to build with Windows. Thank you to
bb53026 
https://stackoverflow.com/questions/78124769/pybind11-how-do-i-truly-fix-missing-dll-error
for the tip on statically linking lpthread.
@leakec
Trying to fix finding the right executable with cibuildwheel.
c59b09b
@leakec
Using Development.Module instread. See notes in pybind11 here: https:…
16b2ad7 
…//pybind11.readthedocs.io/en/stable/compiling.html#findpython-mode
@leakec
Undoing this, as this breaks source builds.
c50a816
@leakec
Letting pybind11 find Python. Hoping this will fix cibuildwheel findi…
025f96d 
…ng the wrong version.
@leakec
Adding ninja to cibuildwheel runners.
586ed2c
@leakec
Adding Wextra and fixing the errors. Still need to fix pedantic, but …
1e4d696 
…need an older version of gcc.
@leakec
Fixing pedantic errors.
6eaf591
@leakec
No need to skip pp*, as these have been deprecated.
244ce6a
@leakec
Some quick typing improvements.
3fac0a8 
More typing fixes.
@leakec
Formatting
501a39c
@leakec
Getting rid of unecessary imports.
b1a726f
@leakec leakec self-assigned this Aug 4, 2025
@leakec leakec merged commit 6535f1c into main Aug 4, 2025
1 check passed
@leakec leakec deleted the jax_0_7 branch August 4, 2025 02:32
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

No reviews

Assignees

@leakec leakec

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

Upgrade to JAX 0.7.0

2 participants

@leakec