Skip to content

Latest commit

 

History

History

flaxlib_src

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

flaxlib

Build flaxlib from source

Install necessary dependencies to build the C++ based package.

pip install meson-python ninja build

Clone the Flax repository, navigate to the flaxlib source directory.

git clone git@github.com:google/flax.git
cd flax/flaxlib_src

Configure the build.

mkdir -p subprojects
meson wrap install robin-map
meson wrap install nanobind
meson setup builddir

Compile the code. You'll need to run this repeatedly if you modify the source code. Note that the actual wheel name will differ depending on your system.

meson compile -C builddir
python -m build . -w
pip install dist/flaxlib-0.0.1-cp311-cp311-macosx_14_0_arm64.whl --force-reinstall