JAX implementation of Large Language Models. You can train GPT-2-like model with 青空文庫 (aozora bunko-clean dataset) or any other text dataset.
☆13Aug 5, 2024Updated last year
Alternatives and similar repositories for jax-llm
Users that are interested in jax-llm are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- [⚠️ WIP] ALMOは拡張Markdownパーサ・静的サイトジェネレータです。WebAssemblyを使ってブラウザ上で完結する実行環境を提供し、サーバを必要としないサンプルコードの実行環境やジャッジシステムを提供するページの構築を可能にします。☆16Feb 28, 2026Updated 3 weeks ago
- Llama-Mimi is a speech language model that uses a unified tokenizer (Mimi) and a single Transformer decoder (Llama) to jointly model sequ…☆30Sep 20, 2025Updated 6 months ago
- YOLOX implemented by pytorch lightning, a simpler expression of pytorch☆11May 26, 2022Updated 3 years ago
- [ICRA 2020] Residual Reactive Navigation: Combining Classical and Learned Navigation Strategies for Deployment in Unknown Environments☆12Jan 23, 2020Updated 6 years ago
- Visualize, share, and keep your favorite music artists on the Web☆14May 23, 2023Updated 2 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click and start building anything your business needs.
- for developing jp-core-v1x series by all members of WG☆13Mar 1, 2026Updated 3 weeks ago
- Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation☆12May 24, 2021Updated 4 years ago
- A library for semantic similarity search☆26Jan 31, 2025Updated last year
- [ICLR 2026] Evaluating the performance of LLMs on Japanese challenging financial tasks.☆31Mar 6, 2026Updated 2 weeks ago
- Easily turn large English text datasets into Japanese text datasets using open LLMs.☆27Jan 20, 2025Updated last year
- An implementation of "Subspace Representations for Soft Set Operations and Sentence Similarities" (NAACL 2024)☆10May 31, 2024Updated last year
- JRDB dataset: trajectory prediction baselines and data preprocessing☆13Nov 7, 2023Updated 2 years ago
- Simple example client demonstrating how to connect to MCP servers over HTTP (SSE)☆17Jan 8, 2025Updated last year
- [wip] Lightweight Automatic Differentiation & DeepLearning Framework implemented in pure Julia.☆30Feb 29, 2024Updated 2 years ago
- Open source password manager - Proton Pass • AdSecurely store, share, and autofill your credentials with Proton Pass, the end-to-end encrypted password manager trusted by millions.
- 法律・判例関係のデータセット☆50Jan 8, 2025Updated last year
- Japanese Programming Language Nadesiko2 (C# version)☆19Nov 2, 2018Updated 7 years ago
- Swallowプロジェクト 大規模言語モデル 評価スクリプト☆24Sep 17, 2025Updated 6 months ago
- Fine-tuning Moshi/J-Moshi on your own spoken dialogue data☆89Jan 5, 2026Updated 2 months ago
- Python implementation of Fast Marching Tree☆16Oct 25, 2020Updated 5 years ago
- Mixtral-based Ja-En (En-Ja) Translation model☆20Jan 6, 2025Updated last year
- Samples for JSON book☆20Apr 15, 2024Updated last year
- ☆27Nov 4, 2024Updated last year
- Implementation of Vision Transformers in Flax☆18Oct 12, 2020Updated 5 years ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- マウスクリックで指定した座標を矩形に射影変換するプログラム。☆10Jul 9, 2020Updated 5 years ago
- 2D gridmap tools☆18Nov 9, 2021Updated 4 years ago
- ☆26Jul 11, 2025Updated 8 months ago
- An interpreter for the Monkey programming language written in C++☆29Jul 12, 2023Updated 2 years ago
- introduction to dataflow analysis using julia☆14Oct 26, 2020Updated 5 years ago
- ☆11Oct 26, 2022Updated 3 years ago
- LLM やシェルとインタラクティブに対話するための macOS 用アプリケーション☆49Mar 22, 2025Updated last year
- 泣き顔に見えて、可哀想...☆10Feb 16, 2023Updated 3 years ago
- ☆11Mar 27, 2021Updated 4 years ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- A library for garbage collection in Rust.☆13Apr 23, 2021Updated 4 years ago
- A Helix plugin to display pressed keys on screen☆31Sep 20, 2025Updated 6 months ago
- Dataset generation scripts for “Path Planning using Neural A* Search” presented in ICML-21☆25Jan 30, 2023Updated 3 years ago
- Unofficial entropix impl for Gemma2 and Llama and Qwen2 and Mistral☆17Jan 12, 2025Updated last year
- ☆21Oct 5, 2019Updated 6 years ago
- JAX implementation of the Mistral 7b v0.1 model☆13Mar 27, 2024Updated last year
- A lightweight framework for evaluating visual-language models.☆41Jan 16, 2026Updated 2 months ago