├── .formatter.exs ├── .github └── workflows │ ├── gh-pages.yml │ └── test.yaml ├── .gitignore ├── CHANGELOG.md ├── LICENSE ├── README.md ├── axon.png ├── examples ├── basics │ ├── multi_input_example.exs │ └── multi_output_example.exs ├── generative │ ├── fashionmnist_autoencoder.exs │ ├── mnist_gan.exs │ └── text_generator.exs ├── structured │ └── credit_card_fraud.exs └── vision │ ├── cifar10.exs │ ├── cnn_image_denoising.exs │ ├── horses_or_humans.exs │ ├── mnist.exs │ └── resnet50.exs ├── guides ├── cheatsheets │ └── axon_pytorch.cheatmd ├── guides.md ├── model_creation │ ├── complex_models.livemd │ ├── custom_layers.livemd │ ├── model_hooks.livemd │ ├── multi_input_multi_output_models.livemd │ ├── sequential_models.livemd │ └── your_first_axon_model.livemd ├── model_execution │ ├── accelerating_axon.livemd │ └── training_and_inference_mode.livemd ├── serialization │ └── onnx_to_axon.livemd └── training_and_evaluation │ ├── custom_models_loss_optimizers.livemd │ ├── instrumenting_loops_with_metrics.livemd │ ├── using_loop_event_handlers.livemd │ ├── writing_custom_event_handlers.livemd │ ├── writing_custom_metrics.livemd │ ├── your_first_evaluation_loop.livemd │ └── your_first_training_loop.livemd ├── lib ├── axon.ex └── axon │ ├── activations.ex │ ├── compiler.ex │ ├── defn.ex │ ├── display.ex │ ├── initializers.ex │ ├── layers.ex │ ├── loop.ex │ ├── loop │ └── state.ex │ ├── loss_scale.ex │ ├── losses.ex │ ├── metrics.ex │ ├── mixed_precision.ex │ ├── mixed_precision │ └── policy.ex │ ├── model_state.ex │ ├── node.ex │ ├── none.ex │ ├── optimizers.ex │ ├── parameter.ex │ ├── quantization.ex │ ├── quantization │ ├── layers.ex │ └── q_tensor.ex │ ├── schedules.ex │ ├── shape.ex │ ├── shared.ex │ ├── stateful_output.ex │ └── updates.ex ├── logo.png ├── mix.exs ├── mix.lock ├── notebooks ├── basics │ └── xor.livemd ├── generative │ ├── fashionmnist_autoencoder.livemd │ ├── fashionmnist_vae.livemd │ └── mnist_autoencoder_using_kino.livemd ├── structured │ └── credit_card_fraud.livemd ├── text │ └── lstm_generation.livemd ├── time_series │ └── apple_stock_prices.livemd └── vision │ ├── horses_or_humans.livemd │ ├── metric-learning.livemd │ └── mnist.livemd ├── sponsors └── dockyard.png └── test ├── axon ├── activations_test.exs ├── compiler_test.exs ├── initializers_test.exs ├── integration_test.exs ├── layers_test.exs ├── loop_test.exs ├── loss_scale_test.exs ├── losses_test.exs ├── metrics_test.exs ├── mixed_precision_test.exs ├── quantization_test.exs └── shape_test.exs ├── axon_test.exs ├── fixtures ├── lstm_cell_test │ ├── test_lstm_cell_bf.npy │ ├── test_lstm_cell_bg.npy │ ├── test_lstm_cell_bi.npy │ ├── test_lstm_cell_bo.npy │ ├── test_lstm_cell_input_c.npy │ ├── test_lstm_cell_input_h.npy │ ├── test_lstm_cell_input_seq.npy │ ├── test_lstm_cell_output_c.npy │ ├── test_lstm_cell_output_h.npy │ ├── test_lstm_cell_whf.npy │ ├── test_lstm_cell_whg.npy │ ├── test_lstm_cell_whi.npy │ ├── test_lstm_cell_who.npy │ ├── test_lstm_cell_wif.npy │ ├── test_lstm_cell_wig.npy │ ├── test_lstm_cell_wii.npy │ └── test_lstm_cell_wio.npy └── lstm_test │ ├── test_lstm_bf.npy │ ├── test_lstm_bg.npy │ ├── test_lstm_bi.npy │ ├── test_lstm_bo.npy │ ├── test_lstm_input_c.npy │ ├── test_lstm_input_h.npy │ ├── test_lstm_input_seq.npy │ ├── test_lstm_output_c.npy │ ├── test_lstm_output_h.npy │ ├── test_lstm_output_seq.npy │ ├── test_lstm_whf.npy │ ├── test_lstm_whg.npy │ ├── test_lstm_whi.npy │ ├── test_lstm_who.npy │ ├── test_lstm_wif.npy │ ├── test_lstm_wig.npy │ ├── test_lstm_wii.npy │ └── test_lstm_wio.npy ├── support ├── axon_test_util.ex └── cases │ └── axon_case.ex └── test_helper.exs /.formatter.exs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/.formatter.exs -------------------------------------------------------------------------------- /.github/workflows/gh-pages.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/.github/workflows/gh-pages.yml -------------------------------------------------------------------------------- /.github/workflows/test.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/.github/workflows/test.yaml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/.gitignore -------------------------------------------------------------------------------- /CHANGELOG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/CHANGELOG.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/README.md -------------------------------------------------------------------------------- /axon.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/axon.png -------------------------------------------------------------------------------- /examples/basics/multi_input_example.exs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/examples/basics/multi_input_example.exs -------------------------------------------------------------------------------- /examples/basics/multi_output_example.exs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/examples/basics/multi_output_example.exs -------------------------------------------------------------------------------- /examples/generative/fashionmnist_autoencoder.exs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/examples/generative/fashionmnist_autoencoder.exs -------------------------------------------------------------------------------- /examples/generative/mnist_gan.exs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/examples/generative/mnist_gan.exs -------------------------------------------------------------------------------- /examples/generative/text_generator.exs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/examples/generative/text_generator.exs -------------------------------------------------------------------------------- /examples/structured/credit_card_fraud.exs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/examples/structured/credit_card_fraud.exs -------------------------------------------------------------------------------- /examples/vision/cifar10.exs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/examples/vision/cifar10.exs -------------------------------------------------------------------------------- /examples/vision/cnn_image_denoising.exs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/examples/vision/cnn_image_denoising.exs -------------------------------------------------------------------------------- /examples/vision/horses_or_humans.exs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/examples/vision/horses_or_humans.exs -------------------------------------------------------------------------------- /examples/vision/mnist.exs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/examples/vision/mnist.exs -------------------------------------------------------------------------------- /examples/vision/resnet50.exs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/examples/vision/resnet50.exs -------------------------------------------------------------------------------- /guides/cheatsheets/axon_pytorch.cheatmd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/guides/cheatsheets/axon_pytorch.cheatmd -------------------------------------------------------------------------------- /guides/guides.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/guides/guides.md -------------------------------------------------------------------------------- /guides/model_creation/complex_models.livemd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/guides/model_creation/complex_models.livemd -------------------------------------------------------------------------------- /guides/model_creation/custom_layers.livemd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/guides/model_creation/custom_layers.livemd -------------------------------------------------------------------------------- /guides/model_creation/model_hooks.livemd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/guides/model_creation/model_hooks.livemd -------------------------------------------------------------------------------- /guides/model_creation/multi_input_multi_output_models.livemd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/guides/model_creation/multi_input_multi_output_models.livemd -------------------------------------------------------------------------------- /guides/model_creation/sequential_models.livemd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/guides/model_creation/sequential_models.livemd -------------------------------------------------------------------------------- /guides/model_creation/your_first_axon_model.livemd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/guides/model_creation/your_first_axon_model.livemd -------------------------------------------------------------------------------- /guides/model_execution/accelerating_axon.livemd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/guides/model_execution/accelerating_axon.livemd -------------------------------------------------------------------------------- /guides/model_execution/training_and_inference_mode.livemd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/guides/model_execution/training_and_inference_mode.livemd -------------------------------------------------------------------------------- /guides/serialization/onnx_to_axon.livemd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/guides/serialization/onnx_to_axon.livemd -------------------------------------------------------------------------------- /guides/training_and_evaluation/custom_models_loss_optimizers.livemd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/guides/training_and_evaluation/custom_models_loss_optimizers.livemd -------------------------------------------------------------------------------- /guides/training_and_evaluation/instrumenting_loops_with_metrics.livemd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/guides/training_and_evaluation/instrumenting_loops_with_metrics.livemd -------------------------------------------------------------------------------- /guides/training_and_evaluation/using_loop_event_handlers.livemd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/guides/training_and_evaluation/using_loop_event_handlers.livemd -------------------------------------------------------------------------------- /guides/training_and_evaluation/writing_custom_event_handlers.livemd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/guides/training_and_evaluation/writing_custom_event_handlers.livemd -------------------------------------------------------------------------------- /guides/training_and_evaluation/writing_custom_metrics.livemd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/guides/training_and_evaluation/writing_custom_metrics.livemd -------------------------------------------------------------------------------- /guides/training_and_evaluation/your_first_evaluation_loop.livemd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/guides/training_and_evaluation/your_first_evaluation_loop.livemd -------------------------------------------------------------------------------- /guides/training_and_evaluation/your_first_training_loop.livemd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/guides/training_and_evaluation/your_first_training_loop.livemd -------------------------------------------------------------------------------- /lib/axon.ex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/lib/axon.ex -------------------------------------------------------------------------------- /lib/axon/activations.ex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/lib/axon/activations.ex -------------------------------------------------------------------------------- /lib/axon/compiler.ex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/lib/axon/compiler.ex -------------------------------------------------------------------------------- /lib/axon/defn.ex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/lib/axon/defn.ex -------------------------------------------------------------------------------- /lib/axon/display.ex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/lib/axon/display.ex -------------------------------------------------------------------------------- /lib/axon/initializers.ex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/lib/axon/initializers.ex -------------------------------------------------------------------------------- /lib/axon/layers.ex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/lib/axon/layers.ex -------------------------------------------------------------------------------- /lib/axon/loop.ex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/lib/axon/loop.ex -------------------------------------------------------------------------------- /lib/axon/loop/state.ex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/lib/axon/loop/state.ex -------------------------------------------------------------------------------- /lib/axon/loss_scale.ex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/lib/axon/loss_scale.ex -------------------------------------------------------------------------------- /lib/axon/losses.ex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/lib/axon/losses.ex -------------------------------------------------------------------------------- /lib/axon/metrics.ex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/lib/axon/metrics.ex -------------------------------------------------------------------------------- /lib/axon/mixed_precision.ex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/lib/axon/mixed_precision.ex -------------------------------------------------------------------------------- /lib/axon/mixed_precision/policy.ex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/lib/axon/mixed_precision/policy.ex -------------------------------------------------------------------------------- /lib/axon/model_state.ex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/lib/axon/model_state.ex -------------------------------------------------------------------------------- /lib/axon/node.ex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/lib/axon/node.ex -------------------------------------------------------------------------------- /lib/axon/none.ex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/lib/axon/none.ex -------------------------------------------------------------------------------- /lib/axon/optimizers.ex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/lib/axon/optimizers.ex -------------------------------------------------------------------------------- /lib/axon/parameter.ex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/lib/axon/parameter.ex -------------------------------------------------------------------------------- /lib/axon/quantization.ex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/lib/axon/quantization.ex -------------------------------------------------------------------------------- /lib/axon/quantization/layers.ex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/lib/axon/quantization/layers.ex -------------------------------------------------------------------------------- /lib/axon/quantization/q_tensor.ex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/lib/axon/quantization/q_tensor.ex -------------------------------------------------------------------------------- /lib/axon/schedules.ex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/lib/axon/schedules.ex -------------------------------------------------------------------------------- /lib/axon/shape.ex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/lib/axon/shape.ex -------------------------------------------------------------------------------- /lib/axon/shared.ex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/lib/axon/shared.ex -------------------------------------------------------------------------------- /lib/axon/stateful_output.ex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/lib/axon/stateful_output.ex -------------------------------------------------------------------------------- /lib/axon/updates.ex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/lib/axon/updates.ex -------------------------------------------------------------------------------- /logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/logo.png -------------------------------------------------------------------------------- /mix.exs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/mix.exs -------------------------------------------------------------------------------- /mix.lock: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/mix.lock -------------------------------------------------------------------------------- /notebooks/basics/xor.livemd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/notebooks/basics/xor.livemd -------------------------------------------------------------------------------- /notebooks/generative/fashionmnist_autoencoder.livemd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/notebooks/generative/fashionmnist_autoencoder.livemd -------------------------------------------------------------------------------- /notebooks/generative/fashionmnist_vae.livemd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/notebooks/generative/fashionmnist_vae.livemd -------------------------------------------------------------------------------- /notebooks/generative/mnist_autoencoder_using_kino.livemd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/notebooks/generative/mnist_autoencoder_using_kino.livemd -------------------------------------------------------------------------------- /notebooks/structured/credit_card_fraud.livemd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/notebooks/structured/credit_card_fraud.livemd -------------------------------------------------------------------------------- /notebooks/text/lstm_generation.livemd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/notebooks/text/lstm_generation.livemd -------------------------------------------------------------------------------- /notebooks/time_series/apple_stock_prices.livemd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/notebooks/time_series/apple_stock_prices.livemd -------------------------------------------------------------------------------- /notebooks/vision/horses_or_humans.livemd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/notebooks/vision/horses_or_humans.livemd -------------------------------------------------------------------------------- /notebooks/vision/metric-learning.livemd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/notebooks/vision/metric-learning.livemd -------------------------------------------------------------------------------- /notebooks/vision/mnist.livemd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/notebooks/vision/mnist.livemd -------------------------------------------------------------------------------- /sponsors/dockyard.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/sponsors/dockyard.png -------------------------------------------------------------------------------- /test/axon/activations_test.exs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/axon/activations_test.exs -------------------------------------------------------------------------------- /test/axon/compiler_test.exs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/axon/compiler_test.exs -------------------------------------------------------------------------------- /test/axon/initializers_test.exs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/axon/initializers_test.exs -------------------------------------------------------------------------------- /test/axon/integration_test.exs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/axon/integration_test.exs -------------------------------------------------------------------------------- /test/axon/layers_test.exs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/axon/layers_test.exs -------------------------------------------------------------------------------- /test/axon/loop_test.exs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/axon/loop_test.exs -------------------------------------------------------------------------------- /test/axon/loss_scale_test.exs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/axon/loss_scale_test.exs -------------------------------------------------------------------------------- /test/axon/losses_test.exs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/axon/losses_test.exs -------------------------------------------------------------------------------- /test/axon/metrics_test.exs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/axon/metrics_test.exs -------------------------------------------------------------------------------- /test/axon/mixed_precision_test.exs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/axon/mixed_precision_test.exs -------------------------------------------------------------------------------- /test/axon/quantization_test.exs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/axon/quantization_test.exs -------------------------------------------------------------------------------- /test/axon/shape_test.exs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/axon/shape_test.exs -------------------------------------------------------------------------------- /test/axon_test.exs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/axon_test.exs -------------------------------------------------------------------------------- /test/fixtures/lstm_cell_test/test_lstm_cell_bf.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/fixtures/lstm_cell_test/test_lstm_cell_bf.npy -------------------------------------------------------------------------------- /test/fixtures/lstm_cell_test/test_lstm_cell_bg.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/fixtures/lstm_cell_test/test_lstm_cell_bg.npy -------------------------------------------------------------------------------- /test/fixtures/lstm_cell_test/test_lstm_cell_bi.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/fixtures/lstm_cell_test/test_lstm_cell_bi.npy -------------------------------------------------------------------------------- /test/fixtures/lstm_cell_test/test_lstm_cell_bo.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/fixtures/lstm_cell_test/test_lstm_cell_bo.npy -------------------------------------------------------------------------------- /test/fixtures/lstm_cell_test/test_lstm_cell_input_c.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/fixtures/lstm_cell_test/test_lstm_cell_input_c.npy -------------------------------------------------------------------------------- /test/fixtures/lstm_cell_test/test_lstm_cell_input_h.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/fixtures/lstm_cell_test/test_lstm_cell_input_h.npy -------------------------------------------------------------------------------- /test/fixtures/lstm_cell_test/test_lstm_cell_input_seq.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/fixtures/lstm_cell_test/test_lstm_cell_input_seq.npy -------------------------------------------------------------------------------- /test/fixtures/lstm_cell_test/test_lstm_cell_output_c.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/fixtures/lstm_cell_test/test_lstm_cell_output_c.npy -------------------------------------------------------------------------------- /test/fixtures/lstm_cell_test/test_lstm_cell_output_h.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/fixtures/lstm_cell_test/test_lstm_cell_output_h.npy -------------------------------------------------------------------------------- /test/fixtures/lstm_cell_test/test_lstm_cell_whf.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/fixtures/lstm_cell_test/test_lstm_cell_whf.npy -------------------------------------------------------------------------------- /test/fixtures/lstm_cell_test/test_lstm_cell_whg.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/fixtures/lstm_cell_test/test_lstm_cell_whg.npy -------------------------------------------------------------------------------- /test/fixtures/lstm_cell_test/test_lstm_cell_whi.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/fixtures/lstm_cell_test/test_lstm_cell_whi.npy -------------------------------------------------------------------------------- /test/fixtures/lstm_cell_test/test_lstm_cell_who.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/fixtures/lstm_cell_test/test_lstm_cell_who.npy -------------------------------------------------------------------------------- /test/fixtures/lstm_cell_test/test_lstm_cell_wif.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/fixtures/lstm_cell_test/test_lstm_cell_wif.npy -------------------------------------------------------------------------------- /test/fixtures/lstm_cell_test/test_lstm_cell_wig.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/fixtures/lstm_cell_test/test_lstm_cell_wig.npy -------------------------------------------------------------------------------- /test/fixtures/lstm_cell_test/test_lstm_cell_wii.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/fixtures/lstm_cell_test/test_lstm_cell_wii.npy -------------------------------------------------------------------------------- /test/fixtures/lstm_cell_test/test_lstm_cell_wio.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/fixtures/lstm_cell_test/test_lstm_cell_wio.npy -------------------------------------------------------------------------------- /test/fixtures/lstm_test/test_lstm_bf.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/fixtures/lstm_test/test_lstm_bf.npy -------------------------------------------------------------------------------- /test/fixtures/lstm_test/test_lstm_bg.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/fixtures/lstm_test/test_lstm_bg.npy -------------------------------------------------------------------------------- /test/fixtures/lstm_test/test_lstm_bi.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/fixtures/lstm_test/test_lstm_bi.npy -------------------------------------------------------------------------------- /test/fixtures/lstm_test/test_lstm_bo.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/fixtures/lstm_test/test_lstm_bo.npy -------------------------------------------------------------------------------- /test/fixtures/lstm_test/test_lstm_input_c.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/fixtures/lstm_test/test_lstm_input_c.npy -------------------------------------------------------------------------------- /test/fixtures/lstm_test/test_lstm_input_h.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/fixtures/lstm_test/test_lstm_input_h.npy -------------------------------------------------------------------------------- /test/fixtures/lstm_test/test_lstm_input_seq.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/fixtures/lstm_test/test_lstm_input_seq.npy -------------------------------------------------------------------------------- /test/fixtures/lstm_test/test_lstm_output_c.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/fixtures/lstm_test/test_lstm_output_c.npy -------------------------------------------------------------------------------- /test/fixtures/lstm_test/test_lstm_output_h.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/fixtures/lstm_test/test_lstm_output_h.npy -------------------------------------------------------------------------------- /test/fixtures/lstm_test/test_lstm_output_seq.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/fixtures/lstm_test/test_lstm_output_seq.npy -------------------------------------------------------------------------------- /test/fixtures/lstm_test/test_lstm_whf.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/fixtures/lstm_test/test_lstm_whf.npy -------------------------------------------------------------------------------- /test/fixtures/lstm_test/test_lstm_whg.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/fixtures/lstm_test/test_lstm_whg.npy -------------------------------------------------------------------------------- /test/fixtures/lstm_test/test_lstm_whi.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/fixtures/lstm_test/test_lstm_whi.npy -------------------------------------------------------------------------------- /test/fixtures/lstm_test/test_lstm_who.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/fixtures/lstm_test/test_lstm_who.npy -------------------------------------------------------------------------------- /test/fixtures/lstm_test/test_lstm_wif.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/fixtures/lstm_test/test_lstm_wif.npy -------------------------------------------------------------------------------- /test/fixtures/lstm_test/test_lstm_wig.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/fixtures/lstm_test/test_lstm_wig.npy -------------------------------------------------------------------------------- /test/fixtures/lstm_test/test_lstm_wii.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/fixtures/lstm_test/test_lstm_wii.npy -------------------------------------------------------------------------------- /test/fixtures/lstm_test/test_lstm_wio.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/fixtures/lstm_test/test_lstm_wio.npy -------------------------------------------------------------------------------- /test/support/axon_test_util.ex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/support/axon_test_util.ex -------------------------------------------------------------------------------- /test/support/cases/axon_case.ex: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/support/cases/axon_case.ex -------------------------------------------------------------------------------- /test/test_helper.exs: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/elixir-nx/axon/HEAD/test/test_helper.exs --------------------------------------------------------------------------------