README.md

---
library_name: keras-hub
license: gemma
license_link: https://ai.google.dev/gemma/terms
pipeline_tag: text-generation
extra_gated_heading: Access CodeGemma on Hugging Face
extra_gated_prompt: To access CodeGemma on Hugging Face, you’re required to review
  and agree to Google’s usage license. To do this, please ensure you’re logged-in
  to Hugging Face and click below. Requests are processed immediately.
extra_gated_button_content: Acknowledge license
---

# CodeGemma

**Google Model Page**: [CodeGemma](https://ai.google.dev/gemma/docs/codegemma)

This model card corresponds to the 7B base version of the Code Gemma model for usage in keras.

Keras models can be used with JAX, PyTorch or TensorFlow as numerical backends.
JAX, with its support for SPMD model paralellism, is recommended for large models.
For more information: [distributed training with Keras and JAX](https://keras.io/guides/distribution/).

You can find other models in the CodeGemma family here:

|    | Base                                               | Instruct                                                             |
|----|----------------------------------------------------|----------------------------------------------------------------------|
| 2B | [codegemma-1.1-2b-keras](https://huggingface.co/google/codegemma-1.1-2b-keras) |  |
| 7B | [**codegemma-7b-keras**](https://huggingface.co/google/codegemma-7b-keras) | [codegemma-1.1-7b-it-keras](https://huggingface.co/google/codegemma-1.1-7b-it-keras)     |

For more information about the model, visit https://huggingface.co/google/codegemma-7b.

Resources and Technical Documentation
: [Technical Report](https://goo.gle/codegemma)
: [Responsible Generative AI Toolkit](https://ai.google.dev/responsible)

Terms of Use
: [Terms](https://www.kaggle.com/models/google/codegemma/license/consent/verify/huggingface?returnModelRepoId=google/codegemma-7b-keras)

Authors
: Google

## Loading the model

```python
import keras_nlp

gemma_lm = keras_nlp.models.GemmaCausalLM.from_preset("hf://google/codegemma-7b-keras")
```