Mojo within a GPU enabled Docker container?
just that looking at Mojo Dockerfile on Github couldnt find GPU support
2 Replies
reading the docs just seeing that GPU support is currently on the pipeline, wanted to ask if anyone knows of a timeline for this as I am really looking forward to it!
also wanted to ask if XLA support is on the roadmap
No timeline for GPU support sorry, but they're working hard on it. XLA is a compiler that can take Tensorflow and JAX models and lower them to MLIR. Mojo is similar in a way but it's a full programming language that lowers directly to MLIR.