5 Commits

Author SHA1 Message Date
Woosuk Kwon
6d62e4c6aa Add torch to dependencies 2024-04-16 17:06:35 +00:00
Woosuk Kwon
eb0a0466a9 Add JAX requirements 2024-04-16 08:05:54 +00:00
Woosuk Kwon
0d6402ddfd Fix requirements 2024-04-10 07:52:45 +00:00
Woosuk Kwon
6894d3efef Add JAX to requirements.txt 2024-04-01 08:23:59 +00:00
Woosuk Kwon
d148c2ef00 Add requirements 2024-04-01 03:17:43 +00:00