9 projects
autoform
Composable function transformations for LLM programs
fieldflow
Textual IR Transformations
flowfield
Textual IR
kernex
Stencil computations in JAX.
pytreeclass
Visualize, create, and operate on pytrees in the most intuitive way possible.
sepes
Visualize, create, and operate on nested trees in the most intuitive way possible.
sobek
Computer vision for JAX
finitediffx
Finite difference tools in JAX
serket
Finite difference tools in JAX
