model-flops

There are 2 repositories 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.

    Language:Python448162068

  • Mr-TalhaIlyas/Tensorflow-Keras-Model-Profiler

    Tensorflow-Keras Model Profiler: Tells you model's memory requirement, no. of parameters, flops etc.

    Language:Python26163