code: public Trainable put(String key, Trainable value) {
          return bundle.put(key, value);
      }
docs: Puts the trainable in the bundle using a specific key and returns the previous entry or null.
  
  @param key
  @param value
  @return