andreped / super-ml-pets
π’ AI for Super Auto Pets
β31Updated 10 months ago
Related projects β
Alternatives and complementary repositories for super-ml-pets
- OpenAI Gym environment for Super Auto Petsβ14Updated 2 years ago
- Super auto pets engine built with reinforment learning training in mindβ67Updated last year
- This is a port of Mistral-7B model in JAXβ30Updated 4 months ago
- A collection of graph neural networks implementations in JAXβ31Updated 11 months ago
- [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivativeβ¦β17Updated last year
- Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any networkβ46Updated 3 months ago
- Neural Networks for JAXβ83Updated last month
- Exact OU processes with JAXβ34Updated last month
- Implementation of a holodeck, written in Pytorchβ17Updated last year
- E(n)-Equivariant Topological Neural Networksβ23Updated last month
- Graph neural networks in JAX.β67Updated 4 months ago
- Neural Optimal Transport with Lagrangian Costsβ47Updated 3 months ago
- β14Updated last year
- Code for Lie Symmetries SSL paperβ19Updated 7 months ago
- Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"β41Updated last year
- Deep Networks Grok All the Time and Here is Whyβ17Updated 5 months ago
- β25Updated 2 months ago
- This is an easy to understand, simplified, broken-down implementation of Diffusion Models written in PyTorch. The architecture is borroweβ¦β26Updated last year
- β283Updated last month
- Official repository of Implicit Neural Convolutional Kernels for Steerable CNNs, Zhdanov et al.β26Updated 7 months ago
- Code for "Goal-Guided Neural Cellular Automata: Learning to Control Self-Organising Systems"β50Updated 2 years ago
- A Reinforcement Learning-based AI for the strategy autobattler game Super Auto Petsβ9Updated last year
- How to use the Flax Linen API to build a convolutional neural network model and train it for image classification (using TensorFlow Datasβ¦β24Updated last year
- β40Updated 4 months ago
- [ICML 2024] Structure-Aware E(3)-Invariant Molecular Conformer Aggregation Networks.β20Updated this week
- Lightning-like training API for JAX with Flaxβ33Updated 6 months ago
- code for "Neural Conservation Laws A Divergence-Free Perspective".β35Updated last year
- By introducing a differentiable contact model, DiffCoSim extends the applicability of Lagrangian/Hamiltonian-inspired neural networks to β¦β31Updated last year
- Open source code for EigenGame.β28Updated last year
- Jax implementation of x-LSTM: Extended Long Short-Term Memory by Beck et al. (2024)β16Updated 3 months ago