Dongyeong Kim

#75 the onp.add.at is changed into index add so it is enhanced. signed a die

Dataloader problem remained

@jepetolee will solve this issue

Remove the input_elements class so it will be more faster than before

Encoder is slow somehow

input layer has returned, that can be used as encoder

@jadkins99 @swannercjj I guess you should check this issue: https://github.com/google/jax/issues/13672#issuecomment-1515544978 It seems just seeding on jax.random does not make it as deterministic operation on GPU because of the optimisation in...