Commit History

Bugfix: pass device type for autocast.
d675fab

daniel shalem commited on

ran the ruff formatter
c042515

zmelumian commited on

Added NIE on xla device
f5895e7

zmelumian commited on

added cuda as optional
85a3cf8

zmelumian commited on

Feature: Add mixed precision support and direct bfloat16 support.
1940326

daniel shalem commited on

Remove the word "pixart" from code.
d504563

Sapir commited on

Lint: added ruff.
325137b

Sapir commited on

working.
86b1a7e

Sapir commited on

refactor
c811a04

Sapir commited on