Universal Transformer with Memory Tokens. Pure JAX NNX. TPU-ready.
☆13May 12, 2026Updated 5 months ago
Alternatives and similar repositories for utm-jax
Users that are interested in utm-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:
- A collection of reusable, high-performance, well-documented, thorough-tested layers and models in Jax☆23Jun 8, 2025Updated last year
- Official Pytorch Implementation of "Zero-Shot Off-Policy Learning" (ICML 2026)☆25Feb 16, 2026Updated 8 months ago
- ☆15Jun 11, 2026Updated 4 months ago
- This repository contains the code for a workshop on Generative AI using Gemini.☆11Jul 22, 2024Updated 2 years ago
- Minimal copilot for your terminal shell, supports 8+ languages, OpenAI + Bring your own model☆18Sep 12, 2026Updated last month
- 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.
- MIndsDB Extension for Docker Desktop☆12Feb 12, 2024Updated 2 years ago
- Applications leveraged by Statistics Canada, ready to launch on Kubernetes using Helm.☆10Dec 8, 2025Updated 10 months ago
- Topology library for Coq☆12Dec 24, 2015Updated 10 years ago
- GprMax-UI provides a high-level API to run gprMax models along with a set of functions for visualization, analysis and interpreting the r…☆18Jul 3, 2026Updated 3 months ago
- Handy tool that facilitates turning images of math equations and charts into LaTeX code, quickly and easily!☆11Sep 1, 2026Updated last month
- Postman & Chatbot Arena for inference benchmarking.☆15Jun 19, 2025Updated last year
- ☆11May 21, 2026Updated 5 months ago
- ☆13Jul 12, 2024Updated 2 years ago
- A repo for frameworks and tools to let locally run LLMs successfully play Pokemon game☆41Oct 13, 2026Updated 2 weeks 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.
- A Golang replacement for the Kubeflow Jupyter Web APIs / Un remplacement Golang pour les API de Web de Jupyter, partie de Kubeflow☆16Oct 7, 2026Updated 3 weeks ago
- Implements MLP-Mixer (https://arxiv.org/abs/2105.01601) with the CIFAR-10 dataset.☆58Jun 26, 2022Updated 4 years ago
- Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Fl…☆78Aug 17, 2024Updated 2 years ago
- Official code for the transformer LLM experiments in the paper [Evolution Strategies at the Hyperscale](https://arxiv.org/abs/2511.16652)…☆16Apr 14, 2026Updated 6 months ago
- Vehicle control☆12Jun 8, 2019Updated 7 years ago
- FalconCV is an open-source python library that offers developers an interface to interact with some of the most popular computer vision f…☆22Jul 19, 2022Updated 4 years ago
- Code for DASH, a faster alternative to Distributed Shampoo via block stacking.☆13Feb 18, 2026Updated 8 months ago
- A set of Python scripts that makes your experience on TPU better☆56Sep 18, 2025Updated last year
- ☆14Jul 13, 2025Updated last year
- End-to-end encrypted cloud storage - Proton Drive • AdSpecial offer: 40% Off Yearly / 80% Off First Month. Protect your most important files, photos, and documents from prying eyes.
- Toy meta-RL environments for testing algorithms implementations☆10Feb 21, 2024Updated 2 years ago
- Dynamic per-token early exit for LLM inference. Skip layers tokens don't need☆34Sep 18, 2026Updated last month
- 任天堂红白机模拟器☆20Oct 30, 2021Updated 5 years ago
- ☆13Oct 11, 2026Updated 2 weeks ago
- ☆29May 11, 2026Updated 5 months ago
- OpenAPI ↔ Python. To/fro routes, models, and tests. Convert between docstrings, classes, methods, argparse, SQLalchemy, Pydantic, JSON-sc…☆20Updated this week
- Jax/Flax rewrite of Karpathy's nanoGPT☆66Feb 15, 2023Updated 3 years ago
- Highly scalable 2D JAX physics engine.☆72Apr 20, 2026Updated 6 months ago
- Integrates Imbue's Cost Aware pareto-Region Bayesian Search (CARBS) with Weights and Biases (WanDB)☆12Mar 17, 2025Updated last year
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- ☆10Jun 27, 2024Updated 2 years ago
- OpenConjecture, a dataset of mathematics conjectures pulled from papers published to the ArXiv☆19Jul 12, 2026Updated 3 months ago
- Code for the paper "FIRE: Frobenius-Isometry Reinitialization for Balancing the Stability–Plasticity Tradeoff" (ICLR 2026 Oral)☆31Apr 27, 2026Updated 6 months ago
- ☆14Mar 2, 2025Updated last year
- OpenAI gym environments for goal-conditioned and language-conditioned reinforcement learning☆14Jan 27, 2026Updated 9 months ago
- Create melodies with AI☆22Apr 7, 2021Updated 5 years ago
- Contains package that allows converting ROS disparity images to depth images.☆12Mar 23, 2020Updated 6 years ago