more ui tweaks

ui/src/app/jobs/new/SimplifiedJob.tsx

@@ -461,8 +461,6 @@ export default function SimplifiedJob({
 
   return (
     <form onSubmit={handleSubmit} className="space-y-8">
-      <h2 className="text-lg font-semibold text-gray-100">Start training</h2>
-      
       <Card title="Training Setup">
         <div className="grid grid-cols-1 md:grid-cols-2 gap-6">
           <TextInput

ui/src/app/jobs/new/page.tsx

@@ -318,19 +318,21 @@ export default function TrainingForm() {
             </MainContent>
           ) : showAdvancedView ? (
             <MainContent className="pt-28">
-              <div className="px-2 pb-6">
-                <AdvancedJob
-                  jobConfig={jobConfig}
-                  setJobConfig={setJobConfig}
-                  status={status}
-                  handleSubmit={handleSubmit}
-                  runId={runId}
-                  gpuIDs={gpuIDs}
-                  setGpuIDs={setGpuIDs}
-                  gpuList={gpuList}
-                  datasetOptions={datasetOptions}
-                  settings={settings}
-                />
+              <div className="px-2 pb-6 h-full">
+                <section className="flex flex-col relative w-full h-full text-left">
+                  <AdvancedJob
+                    jobConfig={jobConfig}
+                    setJobConfig={setJobConfig}
+                    status={status}
+                    handleSubmit={handleSubmit}
+                    runId={runId}
+                    gpuIDs={gpuIDs}
+                    setGpuIDs={setGpuIDs}
+                    gpuList={gpuList}
+                    datasetOptions={datasetOptions}
+                    settings={settings}
+                  />
+                </section>
               </div>
             </MainContent>
           ) : (

ui/src/components/HFJobsWorkflow.tsx

@@ -712,6 +712,7 @@ export default function HFJobsWorkflow({ jobConfig, onComplete, hackathonEligibl
 
   return (
     <div className="space-y-6">
+      <h2 className="text-lg font-semibold text-gray-100">Start training</h2>
       {/* Progress indicator */}
       <div className="flex items-center justify-between mb-6">
         {(['validate', 'upload', 'submit', 'complete'] as Step[]).map((step, index) => (