Tensor Parallelism with JAX + Shard Map
☆11Sep 29, 2023Updated 2 years ago
Alternatives and similar repositories for transformer_shmap
Users that are interested in transformer_shmap are comparing it to the libraries listed below
Sorting:
- Frechet inception distance (FID) evaluation in JAX☆14May 28, 2024Updated last year
- Einsum-like high-level array sharding API for JAX☆34Jul 16, 2024Updated last year
- ESM2 protein language models in JAX/Flax☆18Oct 10, 2022Updated 3 years ago
- JAX for Graphcore IPU (experimental)☆22Mar 12, 2024Updated last year
- Turn jitted jax functions back into python source code☆23Dec 16, 2024Updated last year
- Minimal yet performant LLM examples in pure JAX☆240Jan 14, 2026Updated last month
- Train very large language models in Jax.☆210Oct 21, 2023Updated 2 years ago
- Recognizing and exploiting conjugacy without a domain-specific language☆36Dec 1, 2019Updated 6 years ago
- OpenFLAM: Framewise Language Audio Model☆95Jan 14, 2026Updated last month
- Jax-based quantum state vector simulator.☆25Jun 28, 2025Updated 8 months ago
- LoRA for arbitrary JAX models and functions☆145Feb 26, 2024Updated 2 years ago
- Minimal but scalable implementation of large language models in JAX☆35Nov 28, 2025Updated 3 months ago
- Using JAX to generate piano music as MIDI☆39Nov 28, 2023Updated 2 years ago
- 기획자와 마케터를 위한 이벤트 댓글 분석 - feat. 인프런 새해 다짐 이벤트☆11Apr 22, 2020Updated 5 years ago
- ☆10Oct 9, 2025Updated 4 months ago
- Visualizing 230 years of US Census data☆12Feb 23, 2020Updated 6 years ago
- Einstein Fields official implementation in JAX.☆73Aug 26, 2025Updated 6 months ago
- Minimal, lightweight JAX implementations of popular models.☆205Updated this week
- [데이콘] 가스공급량 수요예측 모델 개발 대회 3등☆11Apr 12, 2022Updated 3 years ago
- EasyRLHF aims to provide an easy and minimal interface to train aligned language models, using off-the-shelf solutions and datasets☆10Dec 12, 2023Updated 2 years ago
- [TMLR 2025] Stability-Aware Training of Machine Learning Force Fields with Differentiable Boltzmann Estimators☆16Nov 20, 2025Updated 3 months ago
- Reference list of email processing resources; focus on preservation and PII handling☆14Apr 20, 2022Updated 3 years ago
- Official implementation of DrugGEN in PyTorch☆12Oct 27, 2023Updated 2 years ago
- 코로나-19 에 대한 확진/완치/사망 에 대한 국내, 해외 정보를 수집합니다. Data scrapes Covid-19 Confirmed/Cured/Deceases Cases.☆10Jun 6, 2021Updated 4 years ago
- Korean Abstract Meaning Representation (AMR) Corpus☆10Feb 27, 2022Updated 4 years ago
- Creating a simple recommendation system on the Basis of similarity☆11Jun 13, 2018Updated 7 years ago
- ☆15Feb 28, 2023Updated 3 years ago
- PaiNN in jax☆11Jan 14, 2025Updated last year
- ☆15May 11, 2025Updated 9 months ago
- [Findings of ACL-2023] This is the official implementation of On the Difference of BERT-style and CLIP-style Text Encoders.☆14Jun 7, 2023Updated 2 years ago
- JAX Multi-Agent RL, Neuro-Evolution, and A-Life Library☆12Oct 12, 2025Updated 4 months ago
- Flutter Application starter using get ecosystem☆10Jan 19, 2021Updated 5 years ago
- Gym implementation of connector to Deepmind lab☆12Mar 26, 2019Updated 6 years ago
- Everything you need to reproduce "Better plain ViT baselines for ImageNet-1k" in PyTorch, and more☆12Feb 16, 2026Updated 2 weeks ago
- ☆12Feb 9, 2022Updated 4 years ago
- Flash cards chrome extensions☆13Jul 12, 2022Updated 3 years ago
- Probabilistic numerical finite differences. Compute finite difference weights and differentiation matrices on scattered data sites and wi…☆12May 8, 2023Updated 2 years ago
- ☆10Mar 10, 2023Updated 2 years ago
- A CLI tool that downloads the materials of a course hosted on the MET (GUC) website, and organizes the materials into their respective fo…☆10Mar 25, 2022Updated 3 years ago