Transformer creation from scratch using Jax.
☆20Dec 8, 2024Updated last year
Alternatives and similar repositories for transformers
Users that are interested in transformers are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- A GPU-accelerated and differentiable library for connectivity modelling at scale.☆15Oct 30, 2026Updated last week
- Implementation of various equivariant models in JAX☆18Apr 12, 2024Updated 2 years ago
- an Open Collaborative project to explore the implications — theoretical or practical — of the PDE perspective of ConvNets☆21Oct 7, 2023Updated 3 years ago
- Graph Neural Networks with JAX/NNX☆20Aug 12, 2026Updated 2 months ago
- This repository includes a realization of the resilient projection-based consensus actor-critic algorithm that is resilient to adversaria…☆11May 23, 2022Updated 4 years ago
- Deploy open-source AI quickly and easily - Special Bonus Offer • AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- Project Page: Integrating Online Learning and Connectivity Maintenance for Communication-Aware Multi-Robot Coordination (IROS 2024)☆25Apr 25, 2025Updated last year
- This is a port of Mistral-7B model in JAX☆34Jul 1, 2024Updated 2 years ago
- Neural Emulator Architectures in JAX.☆28Aug 25, 2026Updated 2 months ago
- Algebraic multigrid solvers in JAX☆43Aug 27, 2026Updated 2 months ago
- [Project Page] A benchmark suite for autoregressive neural emulation of PDEs, accepted at Neurips 2024.☆10May 11, 2026Updated 5 months ago
- Train text generation model with JavaScript.☆15Jul 14, 2024Updated 2 years ago
- Latest version of IFISS software☆11Dec 27, 2025Updated 10 months ago
- Dynamic Neural Cellular Automata. Used primarily in my PhD research. Efficiently learn cellular automata rules that give desired emergent…☆15May 24, 2024Updated 2 years ago
- A GPU-accelerated differentiable fluid simulator written in JAX.☆11Feb 1, 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.
- SPAGeDi: Spatial Pattern Analysis of Genetic Diversity☆10Nov 27, 2018Updated 7 years ago
- 2D Stable Fluids & 3D Stable Fluids using the Fast Fourier Transformation implemented efficiently in Julia.☆16Jan 29, 2022Updated 4 years ago
- A Review of Sensitivity Methods for Differential Equations☆34Nov 26, 2024Updated last year
- Inverse modelling framework for dynamical systems characterised by complex dynamics.☆13Oct 16, 2025Updated last year
- A simple pseudo-spectral solver for the Direct Numerical Simulation (DNS) of the 3D Taylor-Green Vortex in the Julia programming language☆11Jun 6, 2022Updated 4 years ago
- A variational auto-encoder (VAE) framework with a new type of prior "Variational Mixture of Posteriors" prior, or VampPrior for short.☆10Apr 7, 2021Updated 5 years ago
- Visualisation for 4D Volumes (Space + Time) written in WebGPU and Rust☆46Nov 21, 2025Updated 11 months ago
- Graphs defined by a grid of vertices. Compatible with the Graphs.jl ecosystem☆12Oct 28, 2026Updated last week
- ☆13May 14, 2025Updated last year
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Tutorial on TikZ☆13Nov 15, 2024Updated last year
- neural network for estimating demographic maps from SNPs☆13Oct 13, 2026Updated 3 weeks ago
- Single Particle States - Quantum Mechanics in 2D☆11Apr 23, 2021Updated 5 years ago
- ☆15Jul 20, 2026Updated 3 months ago
- Training methodologies for autoregressive neural operators/emulators in JAX.☆16Jun 1, 2026Updated 5 months ago
- An Automized Partial Differential Equation solver based on JAX☆14Jul 9, 2026Updated 3 months ago
- Cross Atlas Remapping via Optimal Transport☆12Dec 14, 2023Updated 2 years ago
- R package for working with spatial absorbing Markov chains☆15Jun 20, 2026Updated 4 months ago
- Differentiable N-body code (JAX)☆11May 8, 2025Updated last year
- 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.
- Simple tools to mix and match PyTorch and Jax - Get the best of both worlds!☆38Jan 8, 2026Updated 9 months ago
- Awesome Equinox - A curated list of resources of https://github.com/patrick-kidger/equinox☆16Jun 24, 2024Updated 2 years ago
- Optimisation on Diffeomorphisms☆12Feb 17, 2025Updated last year
- GPU-accelerated differentiable sparse linear solver library for JAX☆58Oct 26, 2026Updated last week
- [Neurips 2024] A benchmark suite for autoregressive neural emulation of PDEs. (≥46 PDEs in 1D, 2D, 3D; Differentiable Physics; Unrolled T…☆111Jun 1, 2026Updated 5 months ago
- A Python implementation of isotropic remeshing algorithm based on Open3D library☆15Jan 15, 2024Updated 2 years ago
- An open-source Unreal Engine 5 simulator for heterogeneous UAV–UGV teams in photorealistic, dynamic, large-scale environments.☆43Oct 27, 2026Updated last week