Transformer creation from scratch using Jax.
☆19Dec 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:
- an Open Collaborative project to explore the implications — theoretical or practical — of the PDE perspective of ConvNets☆21Oct 7, 2023Updated 2 years ago
- Neural Emulator Architectures in JAX.☆24Feb 20, 2026Updated last month
- Efficiently Composable Data Augmentation on the GPU with Jax☆42May 16, 2025Updated 10 months ago
- 🦀 High-performance local RAG server in Rust that integrates with Claude Desktop via MCP. Search PDF documents privately using Ollama emb…☆28Jun 11, 2025Updated 9 months ago
- ☆36Jul 11, 2017Updated 8 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.
- Workshop that will take you from Graph Neural Networks (GNNs) to Transformers, architectures which have led to numerous breakthrough achi…☆13Sep 11, 2023Updated 2 years ago
- Train text generation model with JavaScript.☆15Jul 14, 2024Updated last year
- ☆13Aug 2, 2025Updated 7 months 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 3 years ago
- 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 4 years ago
- Cross Atlas Remapping via Optimal Transport☆12Dec 14, 2023Updated 2 years ago
- Website for the book "The Elements of Differentiable Programming".☆14Jul 2, 2025Updated 8 months ago
- Understand what physics/algorithms do transformers learn internally when trained on planetary motion☆39Feb 9, 2026Updated last month
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- ☆16Jul 7, 2025Updated 8 months ago
- Awesome Equinox - A curated list of resources of https://github.com/patrick-kidger/equinox☆16Jun 24, 2024Updated last year
- Evaluating methods for estimating aperiodic activity in electrophysiological data.☆16Sep 24, 2024Updated last year
- ☆11Jan 22, 2021Updated 5 years ago
- [ICLR 2025] On Evluating the Durability of Safegurads for Open-Weight LLMs☆13Jun 20, 2025Updated 9 months ago
- Run Claude Code on Ollama☆37Apr 22, 2025Updated 11 months ago
- ☆17Mar 4, 2025Updated last year
- A Python interface for the Dawn WebGPU engine☆18Jan 10, 2026Updated 2 months ago
- Code repository of the paper "Variational Stochastic Gradient Descent for Deep Neural Networks" published at☆41May 5, 2025Updated 10 months ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click and start building anything your business needs.
- Python version of FastPM☆12Jul 20, 2020Updated 5 years ago
- ☆23Aug 4, 2025Updated 7 months ago
- ☆19Dec 9, 2023Updated 2 years ago
- ☆20Sep 30, 2025Updated 5 months ago
- Open source web based EEG visualizer and analyzer☆28Mar 10, 2026Updated 2 weeks ago
- Implementation of various equivariant models in JAX☆12Apr 12, 2024Updated last year
- Simple optimal control framework for python☆13Aug 16, 2018Updated 7 years ago
- OMNI-P2x: A universal neural network potential for excited states☆12Mar 19, 2026Updated last week
- Dual Octree Graph Networks for Learning Adaptive Volumetric Shape Representations [SIGGRAPH 2022]☆23Dec 6, 2025Updated 3 months ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Auto-differentiated descriptors using Enzyme☆12Apr 2, 2025Updated 11 months ago
- Use SINDY algorithm to discover a dynamical system from coronavirus data☆13Apr 9, 2024Updated last year
- python interface to tetgen☆10Dec 13, 2023Updated 2 years ago
- MSPaint for marimo and other Python notebooks☆24Oct 24, 2025Updated 5 months ago
- Turn jitted jax functions back into python source code☆23Dec 16, 2024Updated last year
- Solve the advection diffusion equations looped into an optimization problem with JAX/autodiff☆14May 8, 2025Updated 10 months ago
- (BESIDE) ``Bridge” Enhanced Signed Directed Network Embedding☆11Dec 22, 2018Updated 7 years ago