Hugging Face's logo

flashstep
/
anlp-assignment2-part1

Model card Files
xet
 Community

YAML Metadata Warning:empty or missing yaml metadata in repo card

Check out the documentation for more information.

ANLP Assignment 2 — Part 1

This repository contains the trained model checkpoints for Part 1 of ANLP Assignment 2.

Model Variants

  • model_variant_1.pt — Dense 2-layer MLP
  • model_variant_2.pt — MoE, 4 experts, Top-1 routing
  • model_variant_3.pt — MoE, 4 experts, Top-2 routing
  • model_variant_4.pt — Shared + Routed MoE
  • model_variant_5.pt — Matched Active-Parameter MoE

The checkpoints are provided in PyTorch .pt format.

See the accompanying report for experimental details and results.

Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support