 Nel Nibcord
|
bc2812238f
Removed ensure_session to clean stuff up. May revisit later
|
hai 8 meses |
|
Nel Nibcord
|
e9971f74ae
Abstract load checkpoint method
|
hai 8 meses |
|
Nel Nibcord
|
223d35cea0
circular include lol
|
hai 8 meses |
|
Nel Nibcord
|
8dc73074e6
Nodes don't need an abstract base class
|
hai 7 meses |
|
Nel Nibcord
|
59af2dd592
Do we need casting here?
|
hai 8 meses |
|
Nel Nibcord
|
b22c21ac16
Some session method cleanup
|
hai 8 meses |
|
Nel Nibcord
|
98edb393b2
Initialize inference engine session in base class
|
hai 8 meses |
|
Nel Nibcord
|
bcf87e79b7
Okay let's turn no_grad back on. We'll worry about that when tinygrad training works
|
hai 8 meses |
|
Nel Nibcord
|
b7bbda3348
Removed tinygrad StatefulModel class, as it's no longer used
|
hai 8 meses |
|
Nel Nibcord
|
67f5ae25a5
Fixing tinygrad model
|
hai 8 meses |
|
Nel Nibcord
|
bfa3b36be5
Fixing tinygrad model
|
hai 8 meses |
|
Nel Nibcord
|
37a75d6b96
Fixing tinygrad model
|
hai 8 meses |
|
Nel Nibcord
|
0d3abfca95
Made models save properly
|
hai 8 meses |
|
Nel Nibcord
|
9283f6d7bd
Correct loss propagation so we can see the actual loss instead of just the requestor shard's loss
|
hai 8 meses |
|
Nel Nibcord
|
9eadee310b
Basic model saving
|
hai 8 meses |
|
Nel Nibcord
|
38e368f00b
Fixed up the ops so that batches work
|
hai 8 meses |
|
Nel Nibcord
|
dd3d99043b
Working distributed training
|
hai 8 meses |
|
Nel Nibcord
|
175ebc1c42
Coordination biz
|
hai 8 meses |
|
Nel Nibcord
|
3e869051f6
Okay we should probably await the update
|
hai 8 meses |
|
Nel Nibcord
|
75c8650f1f
Naive network-propagated loss implementation on MLX
|
hai 8 meses |
|
Nel Nibcord
|
836856824e
WIP: Training works on mlx
|
hai 8 meses |
|
Nel Nibcord
|
a6fd7a3430
Generalizing some of the dataset biz while also creating uniform batches
|
hai 8 meses |
|
Nel Nibcord
|
f5efbe1b8f
Initial distributed evaluation implementation
|
hai 8 meses |
 Alex Cheema
|
db9de97fa6
Merge pull request #549 from exo-explore/fixtokenencode
|
hai 8 meses |
 Alex Cheema
|
c593434808
fix encode endpoint
|
hai 8 meses |
|
Alex Cheema
|
9f86737a94
fix token encode to use the right model
|
hai 8 meses |
|
Alex Cheema
|
d411559a8b
Merge pull request #548 from exo-explore/subprocessforkfix
|
hai 8 meses |
|
Alex Cheema
|
24130da4fd
prio mac check for interface
|
hai 8 meses |
|
Alex Cheema
|
d6d74e9c0e
Merge pull request #547 from exo-explore/subprocessforkfix
|
hai 8 meses |
|
Alex Cheema
|
31d7bc2df6
subprocess fork fix
|
hai 8 meses |
