JAX/Flax implimentation of 'Attention Is All You Need' by Vaswani et al. (https://arxiv.org/abs/1706.03762)
β15Aug 16, 2021Updated 5 years ago
Alternatives and similar repositories for vanilla-transformer-jax
Users that are interested in vanilla-transformer-jax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Nadir: Cutting-edge PyTorch optimizers for simplicity & composability! π₯ππ»β15Jun 15, 2024Updated 2 years ago
- Simple, extensible implementations of some meta-learning algorithms in Jaxβ11Oct 6, 2020Updated 5 years ago
- β18Aug 17, 2024Updated 2 years ago
- Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.β21Sep 18, 2020Updated 5 years ago
- Keras-like APIs for JAX frameworkβ50Mar 25, 2023Updated 3 years ago
- Bare Metal GPUs on DigitalOcean Gradient AI β’ AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- [ICML 2024] Official PyTorch implementation of the Vectorized Conditional Neural Field.β18Aug 1, 2024Updated 2 years ago
- Benchmarking of diffusion models for global field reconstruction from sparse observationsβ33Mar 22, 2026Updated 4 months ago
- minimalistic AI library that resembles HF's transformersβ13Dec 31, 2024Updated last year
- β18Aug 6, 2024Updated 2 years ago
- Practicum on Supervised Learning in Function Spacesβ36Feb 17, 2022Updated 4 years ago
- Code and Dataset release of "Carpe Diem: On the Evaluation of World Knowledge in Lifelong Language Models" (NAACL 2024)β10Oct 16, 2024Updated last year
- A pathway and collection of resources to learning Jax from beginning to advance.β11Jan 2, 2021Updated 5 years ago
- QUESTDB: A Database of Highly-Accurate Excitation Energiesβ26Jul 30, 2026Updated 2 weeks ago
- This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.β14Dec 21, 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 python module and example scripts to perform molecular simulationts with the variational quantum eigensolver (VQE)β29Jul 22, 2022Updated 4 years ago
- PINNs with irreversible constraints.β17Jun 10, 2026Updated 2 months ago
- Procgen2: A community maintained fork of procgenβ12Aug 3, 2026Updated 2 weeks ago
- νκ΅μ΄ λ¬Έμ₯ λΆμ μμ€ν BCD-KL-Parserβ10Jun 23, 2020Updated 6 years ago
- Flexible, general-purpose VMC framework, built on JAX.β32Jun 13, 2026Updated 2 months ago
- Deep NLP 2 (2019.3-5)β10Feb 19, 2019Updated 7 years ago
- Staggered HARd-constrained Physics-Informed Neural Networks for Phase-field modelling of corrosionβ17Feb 19, 2025Updated last year
- Layered distributions using FLAX/JAXβ10Dec 13, 2020Updated 5 years ago
- Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variationβ12May 24, 2021Updated 5 years ago
- Managed hosting for WordPress and PHP on Cloudways β’ AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- β43Jul 22, 2024Updated 2 years ago
- 200,000+ Sentences about Donald Trump with political bias labelsβ17Jun 2, 2020Updated 6 years ago
- Precision Knowledge Editing (PKE): A novel method to reduce toxicity in LLMs while preserving performance, with robust evaluations and haβ¦β12Nov 26, 2024Updated last year
- QLoRA: Efficient Finetuning of Quantized LLMsβ11Jul 22, 2023Updated 3 years ago
- β13Jul 31, 2023Updated 3 years ago
- TANTE: Time-Adaptive Operator Learning via Neural Taylor Expansionβ18Jul 3, 2026Updated last month
- Paper introducing jax-cosmoβ13Apr 27, 2023Updated 3 years ago
- An RPG Maker MZ pluginβ13Nov 2, 2023Updated 2 years ago
- Building HIPAA Compliant End-to-End Encrypted Chat with Stream React Chat and Virgil Securityβ14Jul 17, 2026Updated last month
- Deploy to Railway using AI coding agents - Free Credits Offer β’ AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Pytorch implementation of NASA: NEURAL ARTICULATED SHAPE APPROXIMATIONβ12May 4, 2021Updated 5 years ago
- β15Oct 29, 2019Updated 6 years ago
- Repository containing code for the NAACL 2021 paper (Incorporating External Knowledge to Enhance Tabular Reasoning)β16Jun 20, 2021Updated 5 years ago
- β10Mar 14, 2021Updated 5 years ago
- Code for the blog postβ12Jan 15, 2021Updated 5 years ago
- Web server driven by handlebars templates, native executables, and static filesβ20Jun 23, 2026Updated last month
- Short-time Fourier transform (STFT) for JAXβ15Dec 20, 2021Updated 4 years ago