# A High level training API for Lux?

**URL:** <https://discourse.julialang.org/t/a-high-level-training-api-for-lux/142874>\
**Category:** Machine Learning\
**Tags:** question\
**Created:** [November 11, 2026, 11:54am UTC](https://discourse.julialang.org/t/a-high-level-training-api-for-lux/142874 "2026-11-11T11:54:16Z")\
**Posts on this page:** 1\
**Page:** 1

<div class="post-metadata">

**Author:** ![thimotedupuch](https://sea2.discourse-cdn.com/julialang/user_avatar/discourse.julialang.org/thimotedupuch/32/222492_2.png) [@thimotedupuch](https://discourse.julialang.org/u/thimotedupuch)\
**Post date:** [November 11, 2026, 11:54am UTC](https://discourse.julialang.org/t/a-high-level-training-api-for-lux/142874/1 "2026-11-11T11:54:16Z")

</div>

Lux, especially with Enzyme and Reactant is really good for deep learning research. I am aware there is also ongoing work to make SciML solvers compatible with Reactant which would be super interesting as well.

I also read Carroll Vance’s blog post and learned a lot.

> [@\[Blog\] Replacing PyTorch with Julia and Reactant: One Year In](https://discourse.julialang.org/t/blog-replacing-pytorch-with-julia-and-reactant-one-year-in/140158):
>
> Over the past year, [Medical Metrics Inc.](https://medicalmetrics.com/) has been migrating our internal model training/serving stack to Julia and Reactant. This resulted in several open source projects building on top of Reactant, and I thought it would be a good idea to write a deep dive style blog post on how these came to be and how they fit together, including an end-to-end example starting from a dataset and finishing with calling a served model from Julia over gRPC: The tutorial part of the post covers the following …

I am reading Chollet’s book about Keras3 (the 2025 edition). People would be surprised how far you can go with Keras even for research without having to go down the full PyTorch or JAX way.

As the creator of Pluto.jl said, we should strive to make julia faster and more powerful, but also simpler to learn and use. If we look at how to train a ConvNet or ResNet for image classification on MNIST or CIFAR10 with Lux, the tutorials cover that very well but the code is not exactly straightforward : users have to define very low level functions (some amount of “boilerplate” one could say), write their own training loops, callbacks, etc…

Now if you add GPU, compilation, data movement between processing units, and other things it’s actually hard to get everything right for a beginner friendly training workflow.

Do you think a new package that is a high level training API would be relevant ? What would it look like ?