maxtext icon indicating copy to clipboard operation
maxtext copied to clipboard

Published 20 hours ago • verified publisher icon google
→

Metadata

A simple, performant and scalable Jax LLM!

Results 159 maxtext issues
Sort by recently updated
recently updated
newest added

Anisha jsonl tfrecord

A9isha avatar A9isha

GCS Tessellation - automation benchmark tool

This is the draft branch to track the work for the automation benchmark tool for GCS tessellation and won't be merged into main.

llukai avatar llukai

Llama2 70b host offload

yejingxin avatar yejingxin

Adding FP8 support through NVIDIA Transformer Engine

Opening a draft pull request while we debug some XLA pattern matcher issues.

abhinavgoel95 avatar abhinavgoel95

Make small changes for running it via XPK.

yangyuwei avatar yangyuwei

Loading real data on subset of hosts

khatwanimohit avatar khatwanimohit

Mattdavidow build debug

gobbleturk avatar gobbleturk

[DO NOT MERGE]test gamma only

Only to test gamma in unit test

NinaCai avatar NinaCai

Docker Build Dependency Image

gobbleturk avatar gobbleturk

test runner push permission

gobbleturk avatar gobbleturk

← Metadata

1.3k
Stars
226
Forks
22
Watchers

Owner

verified publisher icon google

Metadata

A simple, performant and scalable Jax LLM!

Back