Build against openxla XLA repo - Githubissues

joelberkeley / spidr

Accelerated machine learning with dependent types

Apache License 2.0

72 stars 4 forks source link

Build against openxla XLA repo #393

Closed joelberkeley closed 2 months ago

joelberkeley commented 5 months ago

BEFORE MERGE

[ ] check pack install actually works including dynamic libs (either create a toy project and add that to pack, or use a fork of pack-db)
[ ] delete releases and tags

commit message

migrate to OpenXLA

use PJRT to compile and execute graph
can now choose the runtime device, via wrapped PJRT plugins
allow users to create and use custom PJRT plugins
remove source dependency on TensorFlow
build all binaries in spidr, rather than using elixir-nx
add build (not test) for CUDA in CI
automatically install support libraries to idris path (no longer global for user)
add tutorial on spidr internals