model-flops

There are 0 repository under model-flops topic.

google / paxml
Pax is a Jax-based machine learning framework for training large scale models. Pax allows for advanced and fully configurable experimentation and parallelization, and has demonstrated industry leading model flop utilization rates.
c4 gpt jax large-language-models llm model-flops parallelism
Language:Python 418
Mr-TalhaIlyas / Tensorflow-Keras-Model-Profiler
Tensorflow-Keras Model Profiler: Tells you model's memory requirement, no. of parameters, flops etc.
gpu tensorflow model-flops model-params gpu-memory-profiler
Language:Python 25

google / paxml