Trains configurable MLPs with Flax NNX for classification and regression tasks.
npx skills add https://github.com/yonesuke/skills --skill training-mlps
Please help me install this Agent Skill. Skill: training-mlps Source: https://github.com/yonesuke/skills/tree/main/training-mlps Command: npx skills add https://github.com/yonesuke/skills --skill training-mlps
Provides a reusable, configurable Multi-Layer Perceptron (MLP) backbone for supervised learning, reducing boilerplate in model definition and training.
Instantiate MultiLayerPerceptron from the examples, provide input dimensions, and run a forward pass to validate the model.
High-intent search queries and answers about installing and using this skill.
To train a deep learning MLP model using Flax NNX, you can instantiate the MultiLayerPerceptron class with configurable depth, width, and activation functions, then execute a forward pass with your supervised learning data.
A modular MLP backbone is a reusable Multi-Layer Perceptron architecture for supervised learning. You need it for classification and regression tasks requiring compact feedforward networks, quick prototyping, or educational experiments.
Yes, you can use optional LayerNorm or RMSNorm and dropout handling for robust MLP training in JAX, allowing you to configure normalization and regularization within your feedforward network architecture.
You configure hidden layers and output dimensions for a feedforward neural network in JAX by parameterizing the MultiLayerPerceptron builder, adjusting layer size, activation functions, and the final linear output layer.
The best way to build a compact MLP for classification tasks in Flax is using a modular MultiLayerPerceptron backbone, which reduces boilerplate model definition while providing customizable depth, width, and normalization options.
MLPs in JAX are limited to small-to-medium network scales and supervised learning tasks like classification and regression, making them unsuitable for large-scale architectures or complex spatial data processing requiring convolutional layers.
yonesukeBuild, train, and manage Flax NNX neural networks with state handling.
Lord1EgyptLoad, run, and fine-tune pre-trained transformer models for NLP, vision, and audio tasks.
nebiusOrchestrate multi-GPU Open Dreamer world-model training on Nebius clusters.
hung-phanPrototype and train neural networks using high-level Keras APIs.
