Downloads · 30 days
0
felafax/gemma-2-2b-it-JAX
gemma-2-2b-it-JAX is a machine learning model from felafax. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as gemma.
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…
Downloads · 30 days
0
Access
Public
Updated Aug 4, 2024
Repo size
4.1 GB
Likes
1
Public
Click a slice to open those files.
Other4.1 GB · 99%
From the Hugging Face model README
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.
You can fine-tune this model using:
Both options support fine-tuning on TPU for optimal performance.
The model weights were copied from KaggleHub: google/gemma-2/flax/gemma2-2b-it
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.
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.