JAX Llama 4 implementation - #307

Open

SamuelMarks wants to merge 1 commit into

meta-llama:mainfrom

SamuelMarks:jax

Commits on Apr 6, 2025

[llama_models/llama4_jax] Start replacing Torch-isms with JAX equivalents ; [requirements.txt] Add JAX specific dependencies ; [.gitignore] Ignore JetBrains & VScode based IDEs
SamuelMarks
committed
640c8ff