K-H-Ismail / torchorthoLinks
Code for Learnable polynomial, trigonometric, and tropical activations
☆11Updated 3 weeks ago
Alternatives and similar repositories for torchortho
Users that are interested in torchortho are comparing it to the libraries listed below
Sorting:
- Code that searches for ribbon bands☆18Updated 8 months ago
- Exact OU processes with JAX☆55Updated 7 months ago
- A long-horizon, sparse-reward math environment for reinforcement learning. Official code repo for "What makes Math problems hard for rein…☆30Updated 2 months ago
- Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX☆43Updated last year
- Brain-Inspired Modular Training (BIMT), a method for making neural networks more modular and interpretable.☆173Updated 2 years ago
- Multiple dispatch over abstract array types in JAX.☆130Updated last week
- Neural Networks and the Chomsky Hierarchy☆210Updated last year
- Named Tensors for Legible Deep Learning in JAX☆207Updated last week
- JAX Arrays for human consumption☆108Updated 3 months ago
- An introduction to theorem proving in Lean for the impatient.☆19Updated 6 months ago
- Implementation of approximate free-energy minimization in PyTorch☆20Updated 4 years ago
- The boundary of neural network trainability is fractal☆217Updated last year
- Hierarchical Associative Memory User Experience☆104Updated 3 months ago
- Neural Networks for JAX☆84Updated last year
- Running Jax in PyTorch Lightning☆112Updated 10 months ago
- Turn jitted jax functions back into python source code☆22Updated 10 months ago
- A package for defining deep learning models using categorical algebraic expressions.☆61Updated last year
- Stencil computations in JAX☆71Updated 2 years ago
- This repository contains the official code for Energy Transformer---an efficient Energy-based Transformer variant for graph classificatio…☆25Updated last year
- 🧱 Modula software package☆287Updated 2 months ago
- Graph neural networks in JAX.☆68Updated last year
- Automatic Functional Differentiation in JAX☆78Updated last month
- ☆33Updated last year
- Computational abilities and efficiency of neural networks☆56Updated 3 months ago
- Loopy belief propagation for factor graphs on discrete variables in JAX☆157Updated last year
- Implementation of Forward Laplacian algorithm in JAX☆87Updated 3 weeks ago
- nanoGPT using Equinox☆13Updated 2 years ago
- Code for "Learning Graph Cellular Automata" (NeurIPS 2021).☆73Updated 3 years ago
- Fine-grained, dynamic control of neural network topology in JAX.☆21Updated 2 years ago
- Diffusion models in PyTorch☆111Updated last month