training-mlps
CommunityTrain modular MLP backbones with Flax NNX.
Authoryonesuke
Version1.0.0
Installs0
System Documentation
What problem does it solve?
Provides a reusable, configurable Multi-Layer Perceptron (MLP) backbone for supervised learning, reducing boilerplate in model definition and training.
Core Features & Use Cases
- Configurable Architecture: build multiple hidden layers with adjustable size, activation, and output dimension.
- Normalization & Regularization: optional layernorm or rmsnorm and dropout support for robust training.
- Practical Scenarios: classification and regression tasks requiring a compact feedforward model, quick prototyping, or educational experiments using Flax NNX.
Quick Start
Instantiate MultiLayerPerceptron from the examples, provide input dimensions, and run a forward pass to validate the model.
Dependency Matrix
Required Modules
flaxjaxjaxlib
Components
scripts
💻 Claude Code Installation
Recommended: Let Claude install automatically. Simply copy and paste the text below to Claude Code.
Please help me install this Skill: Name: training-mlps Download link: https://github.com/yonesuke/skills/archive/main.zip#training-mlps Please download this .zip file, extract it, and install it in the .claude/skills/ directory.
Agent Skills Search Helper
Install a tiny helper to your Agent, search and equip skill from 223,000+ vetted skills library on demand.