9 projects
pytreeclass
Visualize, create, and operate on pytrees in the most intuitive way possible.
autoform
Composable function transformations for text-space programs
fieldflow
Textual IR Transformations
flowfield
Textual IR
kernex
Stencil computations in JAX.
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
