JAIS-6p7b-Chat

JAIS-6p7b-Chat: Optimized for Qualcomm Devices

JAIS 6.7B is a bilingual large language model (LLM) for both Arabic and English developed by Inception, a G42 company in partnership with MBZUAI and Cerebras. This is a 6.7 billion parameter LLM, trained on a dataset containing 141 billion Arabic tokens and 339 billion English/code tokens. The model is based on transformer-based decoder-only (GPT-3) architecture and uses SwiGLU non-linearity. It implements ALiBi position embeddings, enabling the model to extrapolate to long sequence lengths, providing improved context handling and model precision. The JAIS family of models is a comprehensive series of bilingual English-Arabic LLMs. These models are optimized to excel in Arabic while having strong English capabilities.

This is based on the implementation of JAIS-6p7b-Chat found here. This repository contains pre-exported model files optimized for Qualcomm® devices. You can use the Qualcomm® AI Hub Models library to export with custom configurations. More details on model performance across various devices, can be found here.

Qualcomm AI Hub Models uses Qualcomm AI Hub Workbench to compile, profile, and evaluate this model. Sign up to run these models on a hosted Qualcomm® device.

Getting Started

This model is available for purchase. Please contact us to learn more about licensing options.

Getting Started

Due to licensing restrictions, we cannot distribute pre-exported model assets for this model. Use the Qualcomm® AI Hub Models Python library to compile and export the model with your own:

• Custom weights (e.g., fine-tuned checkpoints)
• Custom input shapes
• Target device and runtime configurations

See our repository for JAIS-6p7b-Chat on GitHub for usage instructions.

Model Details

Model Type: Model_use_case.text_generation

Model Stats:

• Input sequence length for Prompt Processor: 128
• Max context length: 2048
• Minimum QNN SDK version required: 2.27.7
• Number of parameters: 6.7B
• Precision: w4a16 + w8a16 (a few layers)
• Response Rate: Rate of response generation after the first response token.
• Supported languages: Arabic (MSA) and English.
• TTFT: Time To First Token is the time it takes to generate the first response token. This is expressed as a range because it varies based on the length of the prompt. The lower bound is for a short prompt (up to 128 tokens, i.e., one iteration of the prompt processor) and the upper bound is for a prompt using the full context length (2048 tokens).
• Use: Initiate conversation with prompt-processor and then token generator for subsequent iterations.

Performance Summary

References

• Jais and Jais-chat: Arabic-Centric Foundation and Instruction-Tuned Open Generative Large Language Models
• Source Model Implementation

Community

• Join our AI Hub Slack community to collaborate, post questions and learn more about on-device AI.
• For questions or feedback please reach out to us.

Usage and Limitations

This model may not be used for or in connection with any of the following applications:

• Accessing essential private and public services and benefits;
• Administration of justice and democratic processes;
• Assessing or recognizing the emotional state of a person;
• Biometric and biometrics-based systems, including categorization of persons based on sensitive characteristics;
• Education and vocational training;
• Employment and workers management;
• Exploitation of the vulnerabilities of persons resulting in harmful behavior;
• General purpose social scoring;
• Law enforcement;
• Management and operation of critical infrastructure;
• Migration, asylum and border control management;
• Predictive policing;
• Real-time remote biometric identification in public spaces;
• Recommender systems of social media platforms;
• Scraping of facial images (from the internet or otherwise); and/or
• Subliminal manipulation