[nnx] add flaxlib #4

	name: Flaxlib - Build and upload to PyPI

	# for testing only:
	on:
	pull_request:
	branches: [main]

	# on:
	# workflow_dispatch:
	# pull_request:
	# push:
	# branches: [main]
	# paths: ['flaxlib/**']
	# release:
	# types: [published]

	jobs:
	build_wheels:
	name: Build wheels on ${{ matrix.os }}
	runs-on: ${{ matrix.os }}
	strategy:
	matrix:
	# macos-13 is an intel runner, macos-14 is apple silicon
	os: [ubuntu-latest, windows-latest, macos-13, macos-14]

	steps:
	- uses: actions/checkout@v4

	- name: Build wheels
	uses: pypa/cibuildwheel@v2.21.0
	with:
	package-dir: './flaxlib'
	output-dir: './flaxlib/wheelhouse'

	- uses: actions/upload-artifact@v4
	with:
	name: cibw-wheels-${{ matrix.os }}-${{ strategy.job-index }}
	path: ./flaxlib/wheelhouse/*.whl

	build_sdist:
	name: Build source distribution
	runs-on: ubuntu-latest
	steps:
	- uses: actions/checkout@v4

	- name: Build sdist
	run: pipx run build --sdist flaxlib

	- uses: actions/upload-artifact@v4
	with:
	name: cibw-sdist
	path: dist/*.tar.gz

	upload_pypi:
	needs: [build_wheels, build_sdist]
	runs-on: ubuntu-latest
	permissions:
	id-token: write
	steps:
	- uses: actions/download-artifact@v4
	with:
	# unpacks all CIBW artifacts into dist/
	pattern: cibw-*
	path: ./flaxlib/dist
	merge-multiple: true

	- name: Build and publish
	env:
	TWINE_USERNAME: __token__
	TWINE_PASSWORD: ${{ secrets.PYPI_PASSWORD }}
	run: \|
	twine upload flaxlib/dist/*

Provide feedback