{"slug": "aws-trainium-frontier-competition-co-design-models-and-kernels-on-purpose-built", "title": "AWS Trainium Frontier competition: Co-design models and kernels on purpose-built AI chips", "summary": "AWS has launched the AWS Trainium Frontier competition, inviting academic and industry labs to co-design language model architectures and custom kernels on its purpose-built Trainium AI chips. Participants will train models from scratch starting from a ~50M parameter baseline, optimizing architecture, optimizer, training loop, and optional NKI kernels under a fixed time and compute budget to minimize validation bits-per-byte and maximize downstream in-context learning capability.", "body_md": "Modern LLM architectures have co-evolved within a single hardware family. The shapes of our attention mechanisms, the structure of our [multilayer perceptrons](https://www.sciencedirect.com/topics/computer-science/multilayer-perceptron) (MLPs), the choice of numerical formats, and even the granularity of parallelism strategies have all been shaped by hardware constraints: warp sizes, [tensor core](https://www.nvidia.com/en-us/data-center/tensor-cores/) geometries, memory hierarchies, and the [kernel abstractions](https://juliagpu.github.io/KernelAbstractions.jl/stable/) those chips expose. When the hardware changes, the efficient frontier of model architectures changes with it. Here we present an opportunity for academic and industry labs to explore this frontier in detail on AWS Trainium.\n\nPurpose-built accelerators like AWS Trainium present a genuinely different design surface. More on-chip [SRAM (SBUF)](https://awsdocs-neuron.readthedocs-hosted.com/en/v2.26.0/general/nki/programming_model.html), explicit software control over data movement and acceleration at the lowest levels, energy-efficient systolic matrix multiplication (matmuls), and a memory hierarchy designed for training and inference-scale data flows. The resulting [TFLOPs-to-memory-bandwidth](https://gigagpu.com/gpu-memory-bandwidth-vs-tflops/) ratio shifts the performance bottleneck profile: key operations that are memory-bound on conventional accelerators may become compute-bound on Trainium, opening design space for architectures that trade additional computation for reduced memory traffic. These hardware differences mean the optimal attention patterns, MLP structures, and parallelism strategies may be fundamentally different.\n\nThe research question is open: What does an optimal model look like when the hardware constraints are fundamentally different? The AWS Trainium Frontier is a competition designed to answer this question empirically: participants train language models from scratch on Trainium, exploring the full design space from model architecture to custom kernels. The core task is training a language model from scratch, starting from a provided ~50M parameter baseline (nanochat-derived, GPT-style dense LLM with [RMSNorm](https://docs.pytorch.org/docs/2.13/generated/torch.nn.modules.normalization.RMSNorm.html), [rotary embeddings](https://krasserm.github.io/2022/12/13/rotary-position-embedding/), and ReLU² MLP). Participants modify everything: architecture, optimizer, training loop, and optionally custom [NKI kernels](https://awsdocs-neuron.readthedocs-hosted.com/en/latest/nki/index.html). The baseline is a starting point, not a ceiling.\n\nThe AWS Trainium Frontier competition rewards full-stack thinking under a fixed time and compute budget. Participants optimize the model architecture, the optimizer, the training loop, and, if they choose, custom hardware kernels. This enables innovation on a combination of objectives: within the allotted training budget, how low can you drive validation bits-per-byte, and how high can you drive downstream in-context learning capability? The fixed budget creates a direct tradeoff between model capacity (better architecture = fewer steps needed) and training throughput (faster kernels = more steps in the same time). The winning solution finds the balance: the most intelligent model trained most efficiently within the time constraint.\n\nFinal submissions find the optimal point on that frontier, and because Trainium's hardware benefits differ from those of existing accelerators, the optimal architectures will differ as well. Be among the first to discover what model architectures look like when designed for a purpose-built AI chip, contributing to a genuinely new area of [machine learning research](https://www.amazon.science/research-areas/machine-learning).\n\nThe Neuron Kernel Interface (NKI), native PyTorch support, and AI-assisted tooling (including [Amazon Bedrock](https://aws.amazon.com/bedrock/) access) give you direct access to Trainium's unique hardware features: the SBUF scratchpad, TensorEngine tiling, and explicit [DMA control](https://www.geeksforgeeks.org/computer-organization-architecture/direct-memory-access-dma-controller-in-computer-architecture/) that standard framework abstractions cannot expose. This is what enables genuinely hardware-native model designs. The entire NKI API surface fits in a weekend, making it equally accessible to both a human writing kernels by hand and an AI agent generating them under human direction.\n\nThe challenge: Exploring the full design space\n\nPhase 1 gives every team a single [Trainium2](https://aws.amazon.com/ai/machine-learning/trainium/) chip and a 30-minute training budget, fast enough to test dozens of hypotheses in a single day. Phase 1 scores on a single number: validation bits-per-byte (val_bpb) after exactly 30 minutes of training on a single Trn2 chip. Lower is better. Any improvement that fits within that wall-clock budget counts, whether it comes from architecture, optimizer, kernel, or all three. Phase 2 promotes the top 10 teams to a full Trainium2 server with a four-hour budget, opening the door to [distributed parallelism](https://docs.pytorch.org/tutorials/intermediate/ddp_tutorial.html) and communication-aware model shaping. Phase 2 adds a second axis — inference performance on CORE, an aggregate score across in-context learning tasks spanning reasoning, comprehension, and world knowledge. Your final score is a 50/50 composite: you need a model that trains efficiently and learns to reason. It's a research arc from, \"Does my idea work?\" to, \"Does my idea scale?”.\n\nParticipants have flexibility in how they improve the model. A better learning rate schedule matters as much as a faster kernel. This rewards the full stack: a novel attention mechanism is only as fast as the kernel that runs it, and the fastest kernel only matters if the architecture knows how to use it. Model size is uncapped: the constraint isn't parameters, it's time on the chip.\n\nYou choose the model architecture that maximizes capability within a fixed training window. This inversion of the usual scaling paradigm is what makes this a more challenging research question, not just an engineering exercise, and it's where the most publishable insights will emerge.\n\nWhat you get\n\n- A complete nanochat-derived training pipeline with\n[Muon](https://github.com/KellerJordan/Muon)+[AdamW](https://docs.pytorch.org/docs/main/generated/torch.optim.adamw.AdamW_class.html)optimizer, ready to run on all NeuronCores - A Trainium-optimized\n[autoresearch](https://github.com/karpathy/autoresearch)framework for AI-assisted experimentation - Full NKI documentation: programming guide, ISA reference, architecture docs, and example kernels\n[Neuron Explorer](https://awsdocs-neuron.readthedocs-hosted.com/en/latest/tools/neuron-explorer/index.html)for comprehensive profiling and performance debugging of NKI kernels- The CORE evaluation harness for inference self-scoring during Phase 2\n- Eligible academic teams can obtain AWS Promotional Credits covering Trainium compute and Amazon Bedrock access\n- Native PyTorch for Neuron with no additional package installation required\n\nWho should compete\n\n- ML architecture and training researchers exploring model designs, optimizers, and training recipes. Familiarity with PyTorch and transformer training is expected; no hardware kernel experience is required to be competitive at the ML layer.\n- ML systems researchers and performance engineers interested in hardware-aware optimization, custom kernels, and the interplay between model design and hardware. Familiarity with\n[CUDA](https://developer.nvidia.com/cuda),[Triton](https://triton-lang.org/main/index.html), or similar kernel programming transfers directly to NKI; no prior Trainium experience is required. - Teams building with AI research agents, using LLMs and automation to run more experiments, write more kernels, and explore more architectures than any single person could. This competition rewards breadth of exploration, making agentic approaches a natural fit.\n- Teams of one to four members are welcome. Strong submissions will likely combine multiple of these perspectives, either within a single team or via AI-assisted workflows that extend a team’s reach across the stack.\n\nWhat's at stake\n\n- Top three finalists present their work at an exclusive Annapurna Labs research event during NeurIPS 2026 in Sydney, Australia, sharing findings with the ML community and\n[AWS AI Chips](https://aws.amazon.com/ai/machine-learning/inferentia/)leadership. Travel and expenses are the finalists’ responsibility - Top 10 team members receive exclusive Neuron team jackets and finalist swag packs.\n- Top three finalists have the opportunity to co-publish findings with Annapurna Labs researchers, contributing to a seminal paper on hardware-native model design.\n- Prize pool: $25,000 (first), $10,000 (second), $5,000 (third).\n\nKey dates\n\n- Aug. 31, 2026: Phase 1 opens; leaderboard goes live\n- Sept. 30, 2026: Phase 1 closes; top 10 announced\n- Oct. 7, 2026: Phase 2 opens on full Trn2 servers for top 10\n- Nov. 4, 2026: Phase 2 closes\n- Nov. 11, 2026: Finalists selected\n- December 6–12, 2026: Finalist presentations at competition workshop in Sydney\n\n[ Register](https://app.smartsheet.com/b/form/019f19eaab4b772bb689d86d346ba558) by Sept 30, 2026. Other eligibility restrictions apply.\n\n[See terms and conditions](https://trainium-frontier.devpost.com/rules).\n\n*The competition is open to the first 100 teams to register. Participants must be 18 or older. AWS employees, interns, and scholars (2025–2026) and their immediate family members are ineligible. Residents of certain countries are excluded; **see full competition terms for details**. Team sizes can be one to four participants.Register today to secure your team’s spot and start building on genuinely new silicon. The frontier is open — come find out what’s on the other side.*", "url": "https://wpnews.pro/news/aws-trainium-frontier-competition-co-design-models-and-kernels-on-purpose-built", "canonical_source": "https://www.amazon.science/news/aws-trainium-frontier-competition-co-design-models-and-kernels-on-purpose-built-ai-chips", "published_at": "2026-08-10 20:23:04+00:00", "updated_at": "2026-08-10 20:49:42.791854+00:00", "lang": "en", "topics": ["ai-research", "ai-chips", "ai-infrastructure", "machine-learning"], "entities": ["AWS", "AWS Trainium", "Neuron Kernel Interface", "Amazon Bedrock", "nanochat"], "alternates": {"html": "https://wpnews.pro/news/aws-trainium-frontier-competition-co-design-models-and-kernels-on-purpose-built", "markdown": "https://wpnews.pro/news/aws-trainium-frontier-competition-co-design-models-and-kernels-on-purpose-built.md", "text": "https://wpnews.pro/news/aws-trainium-frontier-competition-co-design-models-and-kernels-on-purpose-built.txt", "jsonld": "https://wpnews.pro/news/aws-trainium-frontier-competition-co-design-models-and-kernels-on-purpose-built.jsonld"}}