model-flops topic

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...

google

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

Mr-TalhaIlyas