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:
- Neural Emulator Architectures in JAX.☆26Jun 1, 2026Updated last month
- 🦀 High-performance local RAG server in Rust that integrates with Claude Desktop via MCP. Search PDF documents privately using Ollama emb…☆32Jun 11, 2025Updated last year
- ☆36Jul 11, 2017Updated 8 years ago
- Workshop that will take you from Graph Neural Networks (GNNs) to Transformers, architectures which have led to numerous breakthrough achi…☆12Sep 11, 2023Updated 2 years ago
- Train text generation model with JavaScript.☆15Jul 14, 2024Updated last year
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- ☆13Aug 2, 2025Updated 11 months ago
- Training methodologies for autoregressive neural operators/emulators in JAX.☆13Jun 1, 2026Updated last month
- 2D Stable Fluids & 3D Stable Fluids using the Fast Fourier Transformation implemented efficiently in Julia.☆16Jan 29, 2022Updated 4 years ago
- A simple pseudo-spectral solver for the Direct Numerical Simulation (DNS) of the 3D Taylor-Green Vortex in the Julia programming language☆10Jun 6, 2022Updated 4 years ago
- A GPU-accelerated and differentiable library for landscape genetics and ecological connectivity modelling.☆13Jun 4, 2026Updated last month
- A GPU-accelerated differentiable fluid simulator written in JAX.☆11Feb 1, 2021Updated 5 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
- Cross Atlas Remapping via Optimal Transport☆12Dec 14, 2023Updated 2 years ago
- A complete framework for training Large Language Models from scratch☆23Jan 8, 2026Updated 6 months ago
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- Website for the book "The Elements of Differentiable Programming".☆14Jul 2, 2025Updated last year
- ☆17Jul 7, 2025Updated last year
- Awesome Equinox - A curated list of resources of https://github.com/patrick-kidger/equinox☆16Jun 24, 2024Updated 2 years ago
- Dual Octree Graph Networks for Learning Adaptive Volumetric Shape Representations [SIGGRAPH 2022]☆23Jun 6, 2026Updated last month
- ☆12Jan 22, 2021Updated 5 years ago
- ☆14Feb 18, 2024Updated 2 years ago
- [ICLR 2025] On Evluating the Durability of Safegurads for Open-Weight LLMs☆13Jun 20, 2025Updated last year
- Run Claude Code on Ollama☆37Apr 22, 2025Updated last year
- A Python interface for the Dawn WebGPU engine☆19Jan 10, 2026Updated 5 months ago
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- Code repository of the paper "Variational Stochastic Gradient Descent for Deep Neural Networks" published at☆42May 5, 2025Updated last year
- Understand what physics/algorithms do transformers learn internally when trained on planetary motion☆43Feb 9, 2026Updated 5 months ago
- ☆23Aug 4, 2025Updated 11 months ago
- ☆20Sep 30, 2025Updated 9 months ago
- ☆19Dec 9, 2023Updated 2 years ago
- This repository is the official implementation of Topology-Informed Graph Transformer (Choi et al., GRaM Workshop at ICML 2024).☆12Dec 28, 2024Updated last year
- Implementation of various equivariant models in JAX☆19Apr 12, 2024Updated 2 years ago
- Simple optimal control framework for python☆13Aug 16, 2018Updated 7 years ago
- Auto-differentiated descriptors using Enzyme☆12Apr 2, 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.
- Use SINDY algorithm to discover a dynamical system from coronavirus data☆12Apr 9, 2024Updated 2 years ago
- MSPaint for marimo and other Python notebooks☆25Oct 24, 2025Updated 8 months ago
- Turn jitted jax functions back into python source code☆23Dec 16, 2024Updated last year
- Solving the Navier-Stokes Equations in Python 🐍 simply using NumPy.☆16Jan 28, 2022Updated 4 years ago
- Deep network embedding for graph representation learning in signed networks☆14Jul 8, 2020Updated 6 years ago
- sEEG software☆17May 28, 2026Updated last month
- Diffusion models for disordered materials☆20Nov 18, 2025Updated 7 months ago