Skip to content

Some experiments using Jax + Flax for patitioning LLMs. Also includes some functionality for running on SLURM clusters.

Notifications You must be signed in to change notification settings

MChoi-git/flax_llm_experiments

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

9 Commits
 
 
 
 
 
 

Repository files navigation

About

Some experiments using Jax + Flax for patitioning LLMs. Also includes some functionality for running on SLURM clusters.

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published