bhavnicksm / vanilla-transformer-jaxView external linksLinks
JAX/Flax implimentation of 'Attention Is All You Need' by Vaswani et al. (https://arxiv.org/abs/1706.03762)
β15Aug 16, 2021Updated 4 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
Sorting:
- Nadir: Cutting-edge PyTorch optimizers for simplicity & composability! π₯ππ»β14Jun 15, 2024Updated last year
- Official PyTorch implementation of the Vectorized Conditional Neural Field.β16Aug 1, 2024Updated last year
- Model Agnostic Meta Learning (MAML) implemented in Flax, the neural network library for JAX.β21Sep 18, 2020Updated 5 years ago
- Benchmarking of diffusion models for global field reconstruction from sparse observationsβ30Dec 4, 2024Updated last year
- Practicum on Supervised Learning in Function Spacesβ35Feb 17, 2022Updated 4 years ago
- Precision Knowledge Editing (PKE): A novel method to reduce toxicity in LLMs while preserving performance, with robust evaluations and haβ¦β11Nov 26, 2024Updated last year
- A free and open-source GUI tool that simplifies combining multiple code files into one, with automatic labeling and support for various pβ¦β14Jan 3, 2025Updated last year
- Source code and datasets for Circuit Design Completion using GNNs paperβ10Jan 26, 2023Updated 3 years ago
- Learning Generalized Physical Representation from a Few Examplesβ16Feb 9, 2026Updated last week
- Water wave models in one dimensionβ10Jan 4, 2026Updated last month
- β14Mar 3, 2022Updated 3 years ago
- The PyTorch implementation of paper "KERMIT: Knowledge Graph Completion of Enhanced Relation Modeling with Inverse Transformation"β15Jul 4, 2025Updated 7 months ago
- Benchmarking Autoregressive Conditional Diffusion Models for Turbulent Flow Simulationβ109Dec 13, 2024Updated last year
- A pathway and collection of resources to learning Jax from beginning to advance.β11Jan 2, 2021Updated 5 years ago
- β16Mar 14, 2025Updated 11 months ago
- β10Aug 6, 2022Updated 3 years ago
- β11Nov 4, 2012Updated 13 years ago
- β10Mar 14, 2021Updated 4 years ago
- β12Nov 29, 2021Updated 4 years ago
- Code for the blog postβ12Jan 15, 2021Updated 5 years ago
- Frame-independent vector-cloud neural network for nonlocal constitutive modelling on arbitrary grids.β11Oct 31, 2021Updated 4 years ago
- β15Dec 31, 2023Updated 2 years ago
- Implementation of the Hierarchical and Interpretable Skill Acquisition in Multi-task Reinforcement Learning by Tianmin Shu, Caiming Xiongβ¦β11Jun 18, 2018Updated 7 years ago
- Pytorch implementation of NASA: NEURAL ARTICULATED SHAPE APPROXIMATIONβ12May 4, 2021Updated 4 years ago
- Layered distributions using FLAX/JAXβ10Dec 13, 2020Updated 5 years ago
- νκ΅μ΄ λ¬Έμ₯ λΆμ μμ€ν BCD-KL-Parserβ10Jun 23, 2020Updated 5 years ago
- Published in IEEE Trans. Comput. Imag.β10Oct 22, 2024Updated last year
- An introductory course on computational physics taught at the University of Vermontβ11Dec 3, 2021Updated 4 years ago
- Axion field evolution around QCDβ14Updated this week
- Attend - to what matters.β17Feb 22, 2025Updated 11 months ago
- Repository for course material for Indian Knowledge System (IKS)β13Jan 20, 2026Updated 3 weeks ago
- Generate Parametric Ship Hull with a Guided Tabular Diffusion Modelβ14Dec 5, 2024Updated last year
- Implementation of ICML2023 paper : Coarse-to-Fine: a Hierarchical Diffusion Model for Molecule Generation in 3Dβ50Jul 17, 2023Updated 2 years ago
- β11Mar 5, 2024Updated last year
- minimalistic AI library that resembles HF's transformersβ13Dec 31, 2024Updated last year
- β15Jun 25, 2025Updated 7 months ago
- β15Nov 20, 2023Updated 2 years ago
- Keras-like APIs for JAX frameworkβ50Mar 25, 2023Updated 2 years ago
- Stochastic Physics-Informed Neural Networks: A Moment-Matching Framework for Learning Hidden Physics within Stochastic Differential Equatβ¦β14Dec 21, 2021Updated 4 years ago