felafax/gemma-2-2b-it-JAX

Gemma-2 2b-it Flax Unmodified Model Weights

Model Description

This repository contains the unmodified model weights for the Gemma-2 2b-it model in Flax format. Gemma is a family of lightweight, state-of-the-art open models from Google, designed to help developers and researchers create innovative AI solutions.

Fine-tuning

You can fine-tune this model using:

1. Our hosted service at felafax.ai
2. Our open-source repository RoadrunnerX

Both options support fine-tuning on TPU for optimal performance.

Model Source

The model weights were copied from KaggleHub: google/gemma-2/flax/gemma2-2b-it

Terms of Use

Gemma is provided under and subject to the Gemma Terms of Use. For full details, please visit:

https://ai.google.dev/gemma/terms

By using this model, you agree to comply with these terms.

Acknowledgements

We acknowledge Google for creating and releasing the Gemma model family. This repository aims to make these models more accessible to the community while respecting the terms of use set by Google.